Update api_server.py

api_server.py

@@ -33,6 +33,8 @@ if load_type == 'local':
         raise FileNotFoundError(f"Model file not found at {model_path}")
         
     model = YOLO(model_path)
+
+    print("=============== YOLO DONE =============")
     #model.eval()  # 設定模型為推理模式
 elif load_type == 'remote_hub_download':
     from huggingface_hub import hf_hub_download