Update detector.py

detector.py

@@ -6,7 +6,7 @@ import torch.serialization
 class LBWDetector:
     def __init__(self, model_path='best.pt'):
         """Initialize YOLO model with safe globals for PyTorch 2.6+."""
-        with torch.serialization.safe_globals([torch.nn.modules.container.Sequential]):
+        with torch.serialization.safe_globals([torch.nn.modules.container.Sequential, ultralytics.nn.tasks.DetectionModel]):
             self.model = YOLO(model_path)
 
     def detect_objects(self, frame):