Update app.py

app.py

@@ -7,13 +7,17 @@ from PIL import Image
 import sys
 import os
 
-import sys
-sys.path.append('/home/user/app/yolov10')  # Add your local YOLOv10 repo path
+import torch
+import torch.serialization
+from ultralytics import YOLO
+from ultralytics.nn.tasks import YOLOv10DetectionModel
 
-from ultralytics import YOLO  # This now refers to your local repo, not pip
+# Register the custom class for safe loading
+torch.serialization.add_safe_globals({'YOLOv10DetectionModel': YOLOv10DetectionModel})
 
-model = YOLO('/home/user/app/best.pt')  # Load the model
-model.eval()
+# Load the model inside a safe context
+with torch.serialization.safe_globals({'YOLOv10DetectionModel': YOLOv10DetectionModel}):
+    model = YOLO('/home/user/app/best.pt')  # Load your YOLOv10 model
 
 # Frame processing function
 def process_frame(frame):