ning8429 commited on
Commit
7114c54
·
verified ·
1 Parent(s): 61f9cb6

Update api_server.py

Browse files
Files changed (1) hide show
  1. api_server.py +1 -1
api_server.py CHANGED
@@ -27,7 +27,7 @@ if load_type == 'local':
27
  raise FileNotFoundError(f"Model file not found at {model_path}")
28
 
29
  model = YOLO(model_path)
30
- model.eval() # 設定模型為推理模式
31
  elif load_type == 'remote_hub_download':
32
  from huggingface_hub import hf_hub_download
33
 
 
27
  raise FileNotFoundError(f"Model file not found at {model_path}")
28
 
29
  model = YOLO(model_path)
30
+ #model.eval() # 設定模型為推理模式
31
  elif load_type == 'remote_hub_download':
32
  from huggingface_hub import hf_hub_download
33