jhynsoo/worlabel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix: segmentation get_model() parameter 수정

Browse Source
This commit is contained in:
김진현 2024-09-27 15:49:53 +09:00
parent 669562b712
commit 6a000be8d7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ai/app/api/yolo/segmentation.py
View File

@ -35,9 +35,9 @@ async def segmentation_predict(request: PredictRequest):
return response return response
# 모델 로드 # 모델 로드
def get_model(request: PredictRequest): def get_model(project_id:int, model_key:str):
try: try:
return load_segmentation_model(request.project_id, request.m_key) return load_segmentation_model(project_id, model_key)
except Exception as e: except Exception as e:
raise HTTPException(status_code=500, detail="load model exception: " + str(e)) raise HTTPException(status_code=500, detail="load model exception: " + str(e))