Update app.py

app.py

@@ -6,18 +6,39 @@ pipe = pipeline("object-detection" , model = "hustvl/yolos-tiny")
 
 def detect_objects(image):
     results = pipe(image)
+
     draw = ImageDraw.Draw(image)
+    try:
+        font = ImageFont.truetype("arial.ttf", 18)
+    except:
+        font = ImageFont.load_default()
+
+    for r in results:
+        box = r["box"]
+        label = f"{r['label']} {r['score']:.2f}"
+
+        # stil ayarları
+        color = "red"
+        box_width = 4
+        text_padding = 3
 
-    for r in results: 
-        box = r['box']
+        # Bounding Box
         draw.rectangle(
-            [(box['xmin'], box['ymin']) , (box['xmax'],box['ymax'])], outline = "green" , width = 3 
+            [(box["xmin"], box["ymin"]), (box["xmax"], box["ymax"])],
+            outline=color, width=box_width
         )
 
-        draw.text((box['xmin'], box['ymin']), f"{r['label']} {r['score']:.2f}" , fill = "green")
-
-    for r in results: 
-        print(f"{r['label']} ({r['score']:.2f}) - Box: {r['box']}")
+        # Label
+        text_w, text_h = draw.textsize(label, font=font)
+        text_bg = [
+            (box["xmin"], box["ymin"] - text_h - text_padding),
+            (box["xmin"] + text_w + text_padding * 2, box["ymin"])
+        ]
+        draw.rectangle(text_bg, fill=color)
+        draw.text(
+            (box["xmin"] + text_padding, box["ymin"] - text_h - text_padding),
+            label, fill="white", font=font
+        )
 
     return image