Update detector.py

detector.py

@@ -1,11 +1,12 @@
 from ultralytics import YOLO
-import cv2
 from torch.serialization import add_safe_globals
 from ultralytics.nn.tasks import DetectionModel
+import cv2
+
+add_safe_globals([DetectionModel])  # Allow loading YOLOv8 models with full class
 
 class LBWDetector:
     def __init__(self, model_path='best.pt'):
-        add_safe_globals([DetectionModel])
         self.model = YOLO(model_path)
 
     def detect_objects(self, frame):