Merge branch 'ai/feat/classification' into 'ai/develop'

Fix: segmentation get_model() parameter 수정

See merge request s11-s-project/S11P21S002!225
This commit is contained in:
김진현 2024-09-27 15:50:48 +09:00
commit df0a56565e

View File

@ -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))