polejowska commited on
Commit
f928485
1 Parent(s): ac09809

Update visualization.py

Browse files
Files changed (1) hide show
  1. visualization.py +2 -2
visualization.py CHANGED
@@ -55,8 +55,8 @@ def visualize_prediction(
55
  ax.text(
56
  xmin,
57
  ymin,
58
- f"{label}: {score:0.2f}",
59
- fontsize=10,
60
  bbox=dict(facecolor="yellow", alpha=0.5),
61
  )
62
  ax.axis("off")
 
55
  ax.text(
56
  xmin,
57
  ymin,
58
+ f"{score:0.2f}",
59
+ fontsize=8,
60
  bbox=dict(facecolor="yellow", alpha=0.5),
61
  )
62
  ax.axis("off")