Update app.py

app.py

@@ -9,24 +9,26 @@ import os
 
 import sys
 from pathlib import Path
-import torch
 import dill._dill
+import torch
 from torch.serialization import add_safe_globals, safe_globals
+from torch.nn import Sequential  # 🔥 এই লাইনে Sequential কে allow করব
 
-# Add yolov10 directory to Python path
+# Yolov10 import path add
 sys.path.append(str(Path(__file__).resolve().parent / "yolov10"))
 
-# Import your custom YOLOv10 model
+# Import YOLOv10 detection model class
 from ultralytics.nn.tasks import YOLOv10DetectionModel
 from ultralytics import YOLO
 
-# Register the required classes/functions
-add_safe_globals([YOLOv10DetectionModel, dill._dill._load_type])
+# Add trusted globals
+add_safe_globals([YOLOv10DetectionModel, dill._dill._load_type, Sequential])
 
-# Load your trained YOLOv10 model
-with safe_globals([YOLOv10DetectionModel, dill._dill._load_type]):
+# Safely load the model
+with safe_globals([YOLOv10DetectionModel, dill._dill._load_type, Sequential]):
     model = YOLO('/home/user/app/best.pt')
 
+
     
 # Frame processing function
 def process_frame(frame):