Update app.py

app.py

@@ -16,13 +16,6 @@ def detect(image):
                                                             threshold=0.9,
                                                             target_sizes=target_sizes)[0]
 
-    # Bounding box in COCO format:
-    # [x_min, y_min, width, height]
-    
-    # model predicts bounding boxes and corresponding COCO classes
-    #logits = outputs.logits
-    #bboxes = outputs.pred_boxes
-
     draw = ImageDraw.Draw(image)
     
     # label and the count
@@ -36,15 +29,15 @@ def detect(image):
         counts[label_name] += 1
 
         x1, y1, x2, y2 = tuple(box)
-        draw.rectangle((x1, y1, x2, y2), outline="red", width=3)
-        draw.text((x1, y2), label_name, fill="white")
+        draw.rectangle((x1, y1, x2, y2), outline="red", width=2)
+        draw.text((x1, y1), label_name, fill="white")
         
-    return results["labels"], results["scores"], results["boxes"], image
+    return counts, image
 
 demo = gr.Interface(
     fn=detect,
     inputs=[gr.inputs.Image(label="Input image", type="pil")],
-    outputs=["text", "text", "text", "image"], #, gr.Label(num_top_classes=10)],
+    outputs=[gr.Label(num_top_classes=10), "text", "text", "image"], #, gr.Label(num_top_classes=10)],
     title="Object Counts in Image"
 )