Update app.py

app.py

@@ -35,10 +35,10 @@ def predict(image, threshold=0.25, model_id=None):
     #perform inference
     model.conf = threshold
     results = model(image, size=input_size)
-    results.pandas().xyxy[0].value_counts('name')
+    count_result = results.pandas().xyxy[0].value_counts('name')
     numpy_image = results.render()[0]
     output_image = Image.fromarray(numpy_image)
-    return output_image
+    return output_image, count_result
 
 gr.Interface(
     title=app_title,
@@ -49,7 +49,7 @@ gr.Interface(
         gr.Slider(maximum=1, step=0.01, value=0.25),
         gr.Dropdown(models_ids, value=models_ids[-1]),
     ],
-    outputs=gr.Image(type="pil"),
+    outputs=[gr.Image(type="pil"),gr.Textbox(show_label=False)],
     examples=examples,
     cache_examples=True if examples else Fale,
 ).launch(enable_queue=True)