Update app.py

app.py

@@ -12,23 +12,25 @@ from pathlib import Path
 import dill._dill
 import torch
 from torch.serialization import add_safe_globals, safe_globals
-from torch.nn import Sequential  # 🔥 এই লাইনে Sequential কে allow করব
+from torch.nn import Sequential
 
 # Yolov10 import path add
 sys.path.append(str(Path(__file__).resolve().parent / "yolov10"))
 
-# Import YOLOv10 detection model class
+# Import YOLOv10 classes
 from ultralytics.nn.tasks import YOLOv10DetectionModel
+from ultralytics.nn.modules.conv import Conv  # ✅ নতুনটা
 from ultralytics import YOLO
 
 # Add trusted globals
-add_safe_globals([YOLOv10DetectionModel, dill._dill._load_type, Sequential])
+add_safe_globals([YOLOv10DetectionModel, dill._dill._load_type, Sequential, Conv])
 
 # Safely load the model
-with safe_globals([YOLOv10DetectionModel, dill._dill._load_type, Sequential]):
+with safe_globals([YOLOv10DetectionModel, dill._dill._load_type, Sequential, Conv]):
     model = YOLO('/home/user/app/best.pt')
 
 
+
     
 # Frame processing function
 def process_frame(frame):