Update app.py

app.py

@@ -13,17 +13,18 @@ model = YOLO(model_path)
 def detect_cheerios(image):
     # Run inference on the input image
     results = model(image)
+    image = results[0].plot() 
     
-    # Get the first result (assuming single image input)
-    result = results[0]
+    # # Get the first result (assuming single image input)
+    # result = results[0]
     
-    # Draw bounding boxes on the image
-    for box in result.boxes:
-        x1, y1, x2, y2 = box.xyxy[0]
-        conf = box.conf[0]
-        cv2.rectangle(image, (int(x1), int(y1)), (int(x2), int(y2)), (0, 255, 0), 2)
-        cv2.putText(image, f'Apple: {conf:.2f}', (int(x1), int(y1) - 10),
-                    cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
+    # # Draw bounding boxes on the image
+    # for box in result.boxes:
+    #     x1, y1, x2, y2 = box.xyxy[0]
+    #     conf = box.conf[0]
+    #     cv2.rectangle(image, (int(x1), int(y1)), (int(x2), int(y2)), (0, 255, 0), 2)
+    #     cv2.putText(image, f'Apple: {conf:.2f}', (int(x1), int(y1) - 10),
+    #                 cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
     
     return image