Update models/detection/detector.py

models/detection/detector.py

@@ -28,12 +28,12 @@ class ObjectDetector:
         )
 
     def load_model(self):
-    if self.model is None:
-        from ultralytics import YOLO  # Deferring import to avoid early CUDA trigger
-        self.model = YOLO(self.weights_path)
-        if self.device == "cuda" and torch.cuda.is_available():
-            self.model.to("cuda")  
-    return self
+        if self.model is None:
+            from ultralytics import YOLO  # Deferring import to avoid early CUDA trigger
+            self.model = YOLO(self.weights_path)
+            if self.device == "cuda" and torch.cuda.is_available():
+                self.model.to("cuda")  
+        return self
 
 
     def predict(self, image: Image.Image, conf_threshold=0.25):