Update models/detection/detector.py

models/detection/detector.py

@@ -56,6 +56,7 @@ class ObjectDetector:
             if self.device == "cuda" and torch.cuda.is_available():
                 self.model.to("cuda")
             logger.info(f"✅ YOLO model initialized on {self.device}")
+        return self
 
     def predict(self, image: Image.Image, conf_threshold=0.25):
         self.load_model()