Update app.py

app.py

@@ -33,18 +33,22 @@ def annotate_image(image):
         cv2.rectangle(image, (left, top), (right, bottom), (0, 0, 255), 2)
         cv2.putText(image, f"{prediction['confidence']:.2f}", (left, top - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
     
+    # Get the number of detections
+    detection_count = len(result['predictions'])
+    
     # Convert the image back to RGB for display in Gradio
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-    return image
+    
+    return image, f"Number of detections: {detection_count}"
 
 # Create the Gradio interface
 iface = gr.Interface(
     fn=annotate_image,
     inputs=gr.Image(type="pil"),
-    outputs="image",
+    outputs=["image", "text"],
     title="TB-Bacillus Detection",
     description="Upload an image to get annotated results from the model."
 )
 
 # Launch the app
-iface.launch()
+iface.launch()