Update app.py

app.py

@@ -3,6 +3,9 @@ import torch
 import cv2
 from ultralytics import YOLO
 
+# Add necessary safe globals to avoid the UnpicklingError
+torch.serialization.add_safe_globals([torch, "ultralytics.nn.tasks.DetectionModel"])
+
 # Load YOLO models
 model_yolo11 = YOLO('./data/yolo11n.pt')
 model_best = YOLO('./data/best.pt')