app.py

app.py

@@ -1,14 +1,15 @@
 import torch
+from ultralytics import YOLO
 from torch.serialization import add_safe_globals
 from ultralytics.nn.tasks import DetectionModel
 
-# --- FIX for PyTorch 2.6 ---
-add_safe_globals([DetectionModel])
 
-from ultralytics import YOLO
+torch.serialization.add_safe_globals([torch.nn.modules.container.Sequential])
+
+model = YOLO("yolov8n.pt")
+
+# your other code continues...
 
-# Now safe to load the model
-model = YOLO("yolov8n")
 
 
 def track_objects(video_input):