Update app.py

app.py

@@ -57,8 +57,9 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
         mask = detection.mask
         color = (255, 255, 255)  # White color for bounding boxes and text
 
-        # Draw the bounding box with a white border and filling
-        cv2.rectangle(image_cv2, (box.xmin, box.ymin), (box.xmax, box.ymax), color, -1)  # Filled white rectangle
+        # Draw the filled rectangle (white filling)
+        cv2.rectangle(image_cv2, (box.xmin, box.ymin), (box.xmax, box.ymax), color, cv2.FILLED)  # Filled white rectangle
+        # Draw the border
         cv2.rectangle(image_cv2, (box.xmin, box.ymin), (box.xmax, box.ymax), color, 2)  # Border
 
         cv2.putText(image_cv2, f'{label}: {score:.2f}', (box.xmin, box.ymin - 10),
@@ -219,4 +220,4 @@ gr.Interface(
     outputs=[gr.Image(type="numpy"), gr.Textbox()],
     title="InsectSAM 🐞",
     examples=examples
-).launch()
+).launch()