diff --git a/main.py b/main.py index 950d5bc..8c08025 100644 --- a/main.py +++ b/main.py @@ -6,23 +6,22 @@ from clearml import Task from start import run - -def init_acai_task(task: Task, output: dict) -> None: +def upload_acai_task(task: Task, output: dict) -> None: """UI output 정의에 맞춰 ClearML 아티팩트를 업로드한다.""" result_data = {"output": output, "status": "PASS"} task.upload_artifact(name="final_result", artifact_object=result_data) if __name__ == "__main__": - parser = argparse.ArgumentParser() - parser.add_argument("--image_url", type=str, required=True) - args = parser.parse_args() - task = Task.init( project_name="Normal_Object_Detection", task_name="model-yolo26-human", ) + + parser = argparse.ArgumentParser() + parser.add_argument("--image_url", type=str, required=True) + args = parser.parse_args() task.connect(parser) output = run(args.image_url) - init_acai_task(task, output) + upload_acai_task(task, output)