Update app.py

app.py

@@ -1,10 +1,11 @@
 import gradio as gr
 import torch
 from ultralytics.nn.tasks import DetectionModel
-from torch.nn.modules.container import Sequential  # Allow Sequential if needed
+from torch.nn.modules.container import Sequential
+from ultralytics.nn.modules import Conv  # Import Conv
 
-# Whitelist the globals to bypass the pickle error (only do this if you trust the model source!)
-torch.serialization.add_safe_globals([DetectionModel, Sequential])
+# Whitelist the globals to bypass the pickle errors (only do this if you trust the model source!)
+torch.serialization.add_safe_globals([DetectionModel, Sequential, Conv])
 
 from ultralyticsplus import YOLO
 from PIL import Image