Sompote commited on
Commit
99b3da4
·
verified ·
1 Parent(s): c354b1a

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ def initialize_models():
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
 
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