Spaces:
Sleeping
Sleeping
Update api_server.py
Browse files- 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 |
|