Update app.py

app.py

@@ -10,21 +10,22 @@ import os
 import sys
 from pathlib import Path
 import torch
+import dill._dill
 from torch.serialization import add_safe_globals, safe_globals
 
 # Add yolov10 directory to Python path
 sys.path.append(str(Path(__file__).resolve().parent / "yolov10"))
 
-# Now import from yolov10
-from ultralytics import YOLO
+# Import your custom YOLOv10 model
 from ultralytics.nn.tasks import YOLOv10DetectionModel
+from ultralytics import YOLO
 
-# Register the actual YOLOv10DetectionModel class for safe loading
-add_safe_globals([YOLOv10DetectionModel])
+# Register the required classes/functions
+add_safe_globals([YOLOv10DetectionModel, dill._dill._load_type])
 
-# Load the model safely using context manager
-with safe_globals([YOLOv10DetectionModel]):
-    model = YOLO('/home/user/app/best.pt')  # Make sure this file exists
+# Load your trained YOLOv10 model
+with safe_globals([YOLOv10DetectionModel, dill._dill._load_type]):
+    model = YOLO('/home/user/app/best.pt')
 
     
 # Frame processing function