Update app.py

app.py

@@ -4,17 +4,10 @@ from ultralytics import YOLO
 
 model = YOLO("best.pt")
 
-def greet(detection):
+def greet(source):
 
-    results = model(detection)  # return a list of Results objects
+    result = model(source)  # 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")