From 6a000be8d7d95370d66c0028b7f14cb4e31de0f3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=EA=B9=80=EC=A7=84=ED=98=84?= Date: Fri, 27 Sep 2024 15:49:53 +0900 Subject: [PATCH] =?UTF-8?q?Fix:=20segmentation=20get=5Fmodel()=20parameter?= =?UTF-8?q?=20=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ai/app/api/yolo/segmentation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ai/app/api/yolo/segmentation.py b/ai/app/api/yolo/segmentation.py index 89458ed..4ef0078 100644 --- a/ai/app/api/yolo/segmentation.py +++ b/ai/app/api/yolo/segmentation.py @@ -35,9 +35,9 @@ async def segmentation_predict(request: PredictRequest): return response # 모델 로드 -def get_model(request: PredictRequest): +def get_model(project_id:int, model_key:str): try: - return load_segmentation_model(request.project_id, request.m_key) + return load_segmentation_model(project_id, model_key) except Exception as e: raise HTTPException(status_code=500, detail="load model exception: " + str(e))