Update models/detection/detector.py

models/detection/detector.py

@@ -29,13 +29,23 @@ class ObjectDetector:
 
     def load_model(self):
         if self.model is None:
-            from ultralytics import YOLO  # Deferring import to avoid early CUDA trigger
+            import torch  # Safe to import here
+            from ultralytics import YOLO  # Defer import
+    
+            # FORCE: Avoid triggering CUDA by setting env var
+            os.environ["CUDA_VISIBLE_DEVICES"] = "-1" if self.device == "cpu" else ""
+    
+            # Initialize model
             self.model = YOLO(self.weights_path)
+    
+            # Move to CUDA only if necessary and safe
             if self.device == "cuda" and torch.cuda.is_available():
-                self.model.to("cuda")  
+                self.model.to("cuda")
+    
         return self
 
 
+
     def predict(self, image: Image.Image, conf_threshold=0.25):
         self.load_model()
         results = self.model(image)