Update app.py

app.py

@@ -23,10 +23,11 @@ def snap(image, model, conf, iou):
     elif model == "S-Raw":
         results = s_raw_model(image, conf=conf, iou=iou)
 
-    
-
     # Convert the results list into an output image
     result = results[0]
+
+    if result.boxes.cls.cpu().numpy().size == 0:
+        return [Image.fromarray(image)]
     classes = result.boxes.cls.cpu().numpy()[0]
     probs = result.boxes.conf.cpu().numpy()[0]
     boxes = result.boxes.xyxy.cpu().numpy()
@@ -38,6 +39,7 @@ def snap(image, model, conf, iou):
     print("-------------------")
     print(boxes)
 
+    print(image)
     for i in range(len(boxes)):
         x1, y1, x2, y2 = boxes[i]
         x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)