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
|