Update app.py

app.py

@@ -19,11 +19,10 @@ from ultralytics import YOLO
 
 # Constants
 ASSETS_DIR = Path(__file__).parent / "assets"
-# Add custom model paths
-MODELS_DIR = Path(__file__).parent.parent.parent
-YOLO11N_MODEL = MODELS_DIR / "yolo11n_results" / "best.pt"
-YOLO11S_MODEL = MODELS_DIR / "yolo11s_results" / "best.pt"
-YOLO11M_MODEL = MODELS_DIR / "yolo11m_results" / "best.pt"
+# Custom model paths - locate them relative to the app.py file
+YOLO11N_MODEL = Path(__file__).parent / "yolo11n_results" / "best.pt"
+YOLO11S_MODEL = Path(__file__).parent / "yolo11s_results" / "best.pt"
+YOLO11M_MODEL = Path(__file__).parent / "yolo11m_results" / "best.pt"
 
 DEMO_VIDEOS = {
     "One Person": ASSETS_DIR / "one-by-one-person-detection.mp4",
@@ -465,9 +464,11 @@ class PeopleDetectionApp:
         with st.sidebar:
             st.header("Settings")
             
-            # Create a mapping for readable model names
+            # Create a mapping for readable model names including custom models
             model_options = {
                 "YOLOv8n (Nano)": "yolov8n.pt",
+                "YOLOv8s (Small)": "yolov8s.pt",
+                "YOLOv8m (Medium)": "yolov8m.pt",
                 "YOLO11n (Custom)": str(YOLO11N_MODEL),
                 "YOLO11s (Custom)": str(YOLO11S_MODEL),
                 "YOLO11m (Custom)": str(YOLO11M_MODEL),
@@ -483,6 +484,10 @@ class PeopleDetectionApp:
             # Get the actual model path from the display name
             model_name = model_options[model_display_name]
             
+            # Log model path for debugging
+            st.write(f"Using model: {model_name}")
+            st.write(f"Model exists: {Path(model_name).exists()}")
+            
             # Detection threshold
             detection_threshold = st.slider(
                 "Detection threshold",