Update detector.py

detector.py

@@ -2,12 +2,12 @@ from ultralytics import YOLO
 import cv2
 import torch
 import torch.serialization
-import ultralytics  # Required to access ultralytics.nn.modules.conv.Conv
+import ultralytics  # Required to access ultralytics-specific classes
 
 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, ultralytics.nn.tasks.DetectionModel, ultralytics.nn.modules.conv.Conv]):
+        with torch.serialization.safe_globals([torch.nn.modules.container.Sequential, ultralytics.nn.tasks.DetectionModel, ultralytics.nn.modules.conv.Conv, torch.nn.modules.conv.Conv2d]):
             self.model = YOLO(model_path)
 
     def detect_objects(self, frame):