PyTorch 2.6’s new “safe deserialization” rule

app.py

@@ -1,10 +1,18 @@
+import torch
 import gradio as gr
 import cv2, os, numpy as np, tempfile, time, json
-from ultralytics import YOLO
 from filterpy.kalman import KalmanFilter
 from scipy.optimize import linear_sum_assignment
 from tqdm import tqdm
 
+# --- 🔧 PyTorch 2.6 safe load fix ---
+import ultralytics.nn.tasks as ultralytics_tasks
+torch.serialization.add_safe_globals([ultralytics_tasks.DetectionModel])
+# -----------------------------------
+
+from ultralytics import YOLO
+
+
 # ---------------------------------------------------------
 # ⚙️ INIT
 # ---------------------------------------------------------