Update app.py

app.py

@@ -4,7 +4,6 @@ from ultralytics import ASSETS, YOLO
 
 model = YOLO("best.pt")
 
-
 def predict_image(img, conf_threshold, iou_threshold):
     results = model.predict(
         source=img,
@@ -15,12 +14,17 @@ def predict_image(img, conf_threshold, iou_threshold):
         imgsz=640,
     )
 
+    # Assuming 'results' has a property or method to get detected labels count
+    # If not directly available, you might need to parse the results accordingly
+    fruits_count = sum(1 for _ in results if _.label == "fruit")  # Example, adjust based on actual results structure
+
+    # Only handling the last result for simplicity, adjust according to your needs
     for r in results:
         im_array = r.plot()
         im = Image.fromarray(im_array[..., ::-1])
-
-    return im
-
+    
+    caption = f"Detected fruits: {fruits_count}"  # Modify this line according to the actual object you're detecting
+    return im, caption
 
 iface = gr.Interface(
     fn=predict_image,
@@ -29,11 +33,10 @@ iface = gr.Interface(
         gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence threshold"),
         gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU threshold")
     ],
-    outputs=gr.Image(type="pil", label="Result"),
+    outputs=[gr.Image(type="pil", label="Result"), gr.Textbox(label="Caption")],
     title="My Yield | 🌱",
     description="Estimate the amount of plants per year",
-
 )
 
 if __name__ == '__main__':
-    iface.launch()
+    iface.launch()