Update app.py

app.py

@@ -1,7 +1,5 @@
 # app.py
 
-# app.py
-
 from transformers import pipeline
 import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
@@ -25,7 +23,6 @@ def get_color(label):
 # Detection function
 def detect_objects(img):
     results = detector(img)
-
     draw = ImageDraw.Draw(img)
     font = ImageFont.load_default()
 
@@ -44,12 +41,12 @@ def detect_objects(img):
 
         # Prepare label with confidence
         label_text = f"{label} ({score:.2f})"
-        text_size = draw.textsize(label_text, font=font)
-        text_background = [box["xmin"], box["ymin"] - text_size[1], box["xmin"] + text_size[0], box["ymin"]]
+        text_bbox = draw.textbbox((box["xmin"], box["ymin"]), label_text, font=font)
+        text_background = [text_bbox[0], text_bbox[1], text_bbox[2], text_bbox[3]]
 
         # Draw background for text
         draw.rectangle(text_background, fill=color)
-        draw.text((box["xmin"], box["ymin"] - text_size[1]), label_text, fill="black", font=font)
+        draw.text((text_bbox[0], text_bbox[1]), label_text, fill="black", font=font)
 
     return img