Spaces:
Sleeping
Sleeping
add_safe_globals
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ from scipy.optimize import linear_sum_assignment
|
|
| 13 |
# 🧠 Safe-load fix for PyTorch 2.6
|
| 14 |
# ------------------------------------------------------------
|
| 15 |
import torch, ultralytics.nn.tasks as ultralytics_tasks
|
| 16 |
-
torch.serialization.add_safe_globals([
|
| 17 |
|
| 18 |
# ------------------------------------------------------------
|
| 19 |
# ⚙️ Model + Config
|
|
|
|
| 13 |
# 🧠 Safe-load fix for PyTorch 2.6
|
| 14 |
# ------------------------------------------------------------
|
| 15 |
import torch, ultralytics.nn.tasks as ultralytics_tasks
|
| 16 |
+
torch.serialization.add_safe_globals([ultralytics_tasks.DetectionModel])
|
| 17 |
|
| 18 |
# ------------------------------------------------------------
|
| 19 |
# ⚙️ Model + Config
|