From 8730b54d2d71cb5852db96be187017b5a795d7bd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=EA=B9=80=EC=A7=84=ED=98=84?= <kimjh9903@naver.com>
Date: Fri, 27 Sep 2024 11:28:52 +0900
Subject: [PATCH] =?UTF-8?q?Fix:=20get=5Fmodel=20parameter=20=EC=88=98?=
 =?UTF-8?q?=EC=A0=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ai/app/api/yolo/detection.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ai/app/api/yolo/detection.py b/ai/app/api/yolo/detection.py
index eee5a66..2517220 100644
--- a/ai/app/api/yolo/detection.py
+++ b/ai/app/api/yolo/detection.py
@@ -111,7 +111,7 @@ async def detection_train(request: TrainRequest):
     dataset_root_path = get_dataset_root_path(request.project_id)
 
     # 모델 로드
-    model = get_model(request)
+    model = get_model(request.project_id, request.m_key)
 
     # 이 값을 학습할때 넣으면 이 카테고리들이 학습됨
     names = list(request.label_map)