Update Space

app.py

@@ -78,15 +78,22 @@ def draw_boxes(image: Image.Image, outputs: dict):
         bottom = ymax * image_height
 
         # Draw bounding box
-        draw.rectangle([left, top, right, bottom], outline="red", width=3)
+        draw.rectangle([left, top, right, bottom], outline="red", width=4)
 
         # Prepare label text
         text = f"{label}: {score:.2f}"
 
+        #Text Box will occupy the top left corner of the bounding box 
+        #Size of the box (and text) will be 10% of the height of the bounding box
+        #And width of the entire bounding box
+
+        text_width = right - left
+        text_height = (bottom - top) // 10
+
         # Calculate text size using textbbox
-        text_bbox = draw.textbbox((0, 0), text, font=font)
-        text_width = text_bbox[2] - text_bbox[0]
-        text_height = text_bbox[3] - text_bbox[1]
+        #text_bbox = draw.textbbox((0, 0), text, font=font)
+        #text_width = text_bbox[2] - text_bbox[0]
+        #text_height = text_bbox[3] - text_bbox[1]
 
         # Calculate label background position
         # Ensure the label box does not go above the image