Update app.py

app.py

@@ -6,14 +6,17 @@ import numpy as np
 from ultralytics import YOLO
 import tempfile
 import torch
+import ultralytics.nn.tasks
 
 # Set Ultralytics config path
 os.environ['YOLO_CONFIG_DIR'] = '/tmp/Ultralytics'
 
 # Custom function to load model with trusted weights
 def load_model_with_trusted_weights(model_path):
-    import ultralytics.nn.tasks
-    with torch.serialization.safe_globals([ultralytics.nn.tasks.DetectionModel]):
+    with torch.serialization.safe_globals([
+        ultralytics.nn.tasks.DetectionModel,
+        torch.nn.modules.container.Sequential
+    ]):
         return YOLO(model_path)
 
 # Load both YOLO models
@@ -107,4 +110,4 @@ with gr.Blocks() as app:
     )
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch()