nickmuchi commited on
Commit
9aa4473
1 Parent(s): 51bb6b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def visualize_prediction(img, output_dict, threshold=0.5, id2label=None):
53
  colors = COLORS * 100
54
  for score, (xmin, ymin, xmax, ymax), label, color in zip(scores, boxes, labels, colors):
55
  ax.add_patch(plt.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin, fill=False, color=color, linewidth=3))
56
- ax.text(xmin, ymin, f"{label}: {score:0.2f}", fontsize=15, bbox=dict(facecolor="yellow", alpha=0.5))
57
  plt.axis("off")
58
  return fig2img(plt.gcf())
59
 
 
53
  colors = COLORS * 100
54
  for score, (xmin, ymin, xmax, ymax), label, color in zip(scores, boxes, labels, colors):
55
  ax.add_patch(plt.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin, fill=False, color=color, linewidth=3))
56
+ ax.text(xmin, ymin, f"{label}: {score:0.2f}", fontsize=25, bbox=dict(facecolor="yellow", alpha=0.5))
57
  plt.axis("off")
58
  return fig2img(plt.gcf())
59