Update models/detection/detector.py

models/detection/detector.py

@@ -30,6 +30,7 @@ class ObjectDetector:
         )
 
     def load_model(self):
+        logger.info(f"Loading model from path: {self.weights_path}")
         if self.model is None:
             import torch  # Safe to import here
             from ultralytics import YOLO  # Defer import
@@ -50,6 +51,8 @@ class ObjectDetector:
 
     def predict(self, image: Image.Image, conf_threshold=0.25):
         self.load_model()
+        if self.model is None:
+        raise RuntimeError("YOLO model not loaded. Call load_model() first.")
         results = self.model(image)
         detections = []
         for r in results: