update app.py

app.py

@@ -4,11 +4,13 @@ import tempfile
 import gradio as gr
 from ultralytics import YOLO
 from ultralytics.nn.tasks import DetectionModel
+from ultralytics.nn.modules.conv import Conv
 
-# Add both necessary classes to the safe globals list
+# Add all the classes we've seen so far to the safe globals list
 torch.serialization.add_safe_globals([
     DetectionModel,
-    torch.nn.modules.container.Sequential
+    torch.nn.modules.container.Sequential,
+    Conv
 ])
 
 # Load the YOLO model