Update app.py

app.py

@@ -79,7 +79,7 @@
 # application.launch()
 
 
-
+    
 import gradio as gr
 from PIL import Image
 from ultralytics import YOLO
@@ -99,8 +99,12 @@ def detect_objects(images):
         all_bboxes.append(boxes)
         all_bboxes2.append(boxes2)
 
-        masks = result.masks.xyn                         
-        sub_arrays = [arr.tolist() for arr in masks]     
+        if result.masks is not None:
+            masks = result.masks.xyn                         
+            sub_arrays = [arr.tolist() for arr in masks]
+        else:
+            sub_arrays = []
+            
         all_segments.append(sub_arrays)
 
     return all_bboxes, all_bboxes2, all_segments