Update app.py

app.py

@@ -27,8 +27,8 @@ def snap(image, model, conf, iou):
 
     # Convert the results list into an output image
     result = results[0]
-    classes = result.boxes.cls.cpu().numpy()
-    probs = result.boxes.conf.cpu().numpy()
+    classes = result.boxes.cls.cpu().numpy()[0]
+    probs = result.boxes.conf.cpu().numpy()[0]
     boxes = result.boxes.xyxy.cpu().numpy()
 
     print("-------------------")