Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
|
@@ -54,7 +54,7 @@ def initialize_models():
|
|
| 54 |
model.eval()
|
| 55 |
|
| 56 |
# Load YOLO model
|
| 57 |
-
yolo_model_path = "
|
| 58 |
if not os.path.exists(yolo_model_path):
|
| 59 |
st.error(f"YOLO model file not found: {yolo_model_path}")
|
| 60 |
return device, model, None
|
|
|
|
| 54 |
model.eval()
|
| 55 |
|
| 56 |
# Load YOLO model
|
| 57 |
+
yolo_model_path = "yolo11s.onnx" # Going up one directory since the app.py is in API22_FEB
|
| 58 |
if not os.path.exists(yolo_model_path):
|
| 59 |
st.error(f"YOLO model file not found: {yolo_model_path}")
|
| 60 |
return device, model, None
|