Update utils/video_processing.py

utils/video_processing.py

@@ -10,25 +10,30 @@ MODEL_PATH = 'models/yolov8_model.pt'
 
 # Check if model file exists
 if not os.path.exists(MODEL_PATH):
-    raise FileNotFoundError(f"YOLO model file not found at {MODEL_PATH}. Please ensure 'yolov8_model.pt' is in the 'models/' directory.")
+    raise FileNotFoundError(
+        f"YOLO model file not found at {MODEL_PATH}. "
+        "Please place a valid YOLOv8 .pt file (e.g., yolov8n.pt) in the 'models/' directory. "
+        "You can download it using scripts/download_yolov8_model.py."
+    )
 
 # Load YOLO model
 try:
     # Load the model using Ultralytics YOLO
     model = YOLO(MODEL_PATH)
 except Exception as e:
-    # If loading fails due to weights_only issue, try manual loading
+    # If loading fails, try manual loading
     try:
         # Manually load the checkpoint with weights_only=False
         checkpoint = torch.load(MODEL_PATH, map_location='cpu', weights_only=False)
-        model = YOLO('yolov8n.yaml')  # Load model architecture from YAML
+        model = YOLO('models/yolov8n.yaml')  # Load model architecture from YAML
         model.load_state_dict(checkpoint['model'].state_dict())  # Load weights
     except Exception as inner_e:
         raise RuntimeError(
             f"Failed to load YOLO model from {MODEL_PATH}: {str(e)}. "
             f"Manual loading also failed: {str(inner_e)}. "
-            "Ensure the model is a valid YOLOv8 .pt file from a trusted source. "
-            "You may need to re-save the model or use a pre-trained model like yolov8n.pt."
+            "The model file may be corrupted or not a valid YOLOv8 .pt file. "
+            "Please replace it with a valid model, e.g., by running scripts/download_yolov8_model.py "
+            "to download yolov8n.pt, or train a custom model using scripts/train_yolov8_model.py."
         )
 
 def track_ball(video_path: str) -> list: