dk - edit

main.py

@@ -10,11 +10,17 @@ from clearml import Task  # 1. ClearML 임포트
 # 방법 A) 가장 최신 트렌드인 NMS-Free 임베디드 특화 모델 로드
 model = YOLO("./weights/yolo26m.pt")  # Medium 크기 가중치 자동 다운로드
 
+
+def init_acai_task(output):
     task = Task.init(
         project_name="Normal_Object_Detection", 
         task_name="model-yolo26-human"
     )
 
+    result_data = {"output": output, "status": "PASS"}
+    task.upload_artifact(name="final_result", artifact_object=result_data)
+
+
 def xyxy_to_xywh(box: dict) -> dict:
     x1 = math.floor(box["x1"])
     y1 = math.floor(box["y1"])
@@ -51,8 +57,7 @@ def main(image_path: str):
     }
     print(json.dumps(output, indent=2, ensure_ascii=False))
 
-    result_data = {"output": output, "status": "PASS"}
-    task.upload_artifact(name="final_result", artifact_object=result_data)
+    init_acai_task(output)
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()