Update app.py

app.py

@@ -11,6 +11,8 @@ from torchvision.utils import draw_bounding_boxes
 import rfdetr.datasets.transforms as T
 from torchvision.ops import box_convert
 
+# Adapted from https://huggingface.co/spaces/rizavelioglu/fashionfail
+
 def _box_yxyx_to_xyxy(boxes: torch.Tensor) -> torch.Tensor:
     """Convert bounding boxes from (y1, x1, y2, x2) format to (x1, y1, x2, y2) format.
 
@@ -115,9 +117,9 @@ def draw_predictions(boxes, labels, scores, img, score_threshold=0.5, font_size=
         colors=colors,
         width=10,
         font_size=30,
-        font="m.ttf",
+        font="arial.ttf",
     )
-    imgs_list.append(img_bbox.permute(1, 2, 0).numpy())  # convert to HWC for Gradio
+    imgs_list.append(img_bbox.permute(1, 2, 0).numpy())
 
     return imgs_list