Update app.py

app.py

@@ -29,11 +29,10 @@ def snap(image, model, conf, iou):
         probs = result.boxes.conf.cpu().numpy()
         boxes = result.boxes.xyxy[0].cpu().numpy()
 
-        for i in range(len(boxes)):
-            x1, y1, x2, y2 = boxes[i]
-            x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
-            cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
-            cv2.putText(image, f"{classes[i]} {probs[i]:.2f}", (x1, y1), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
+        x1, y1, x2, y2 = boxes
+        x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
+        cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
+        cv2.putText(image, f"{classes} {probs:.2f}", (x1, y1), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
 
     # Convert the resulting image to a PIL image
     resulting_image = Image.fromarray(image)