Update app.py

app.py

@@ -5,7 +5,17 @@ from ultralytics import YOLO
 model = YOLO("best.pt")
 
 def greet(detection):
-    return model(detection)
+
+    results = model(detection)  # return a list of Results objects
+    
+    for result in results:
+    boxes = result.boxes  # Boxes object for bounding box outputs
+    masks = result.masks  # Masks object for segmentation masks outputs
+    keypoints = result.keypoints  # Keypoints object for pose outputs
+    probs = result.probs  # Probs object for classification outputs
+    obb = result.obb  # Oriented boxes object for OBB outputs
+    result.show()  # display to screen
+    return result.show()
 
 demo = gr.Interface(fn=greet, inputs="image", outputs="image")
 demo.launch()