Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import pandas as pd
 from transformers import AutoImageProcessor, AutoModelForObjectDetection
 from PIL import Image, ImageDraw
 import torch
@@ -31,8 +32,13 @@ def detect(image):
         x1, y1, x2, y2 = tuple(box)
         draw.rectangle((x1, y1, x2, y2), outline="red", width=2)
         draw.text((x1, y1), label_name, fill="white")
+
+    df = pd.DataFrame({
+        'label': [label for label in counts],
+        'counts': [counts[label] for label in counts]
+    })
         
-    return image, counts, counts
+    return image, df, counts
 
 demo = gr.Interface(
     fn=detect,