Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ COLORS = ["#ff7f7f", "#ff7fbf", "#ff7fff", "#bf7fff",
|
|
| 24 |
fdic = {
|
| 25 |
"family" : "Impact",
|
| 26 |
"style" : "italic",
|
| 27 |
-
"size" :
|
| 28 |
"color" : "yellow",
|
| 29 |
"weight" : "bold"
|
| 30 |
}
|
|
@@ -43,7 +43,7 @@ def get_figure(in_pil_img, in_results):
|
|
| 43 |
w, h = prediction['box']['xmax'] - prediction['box']['xmin'], prediction['box']['ymax'] - prediction['box']['ymin']
|
| 44 |
|
| 45 |
ax.add_patch(plt.Rectangle((x, y), w, h, fill=False, color=selected_color, linewidth=3))
|
| 46 |
-
|
| 47 |
|
| 48 |
plt.axis("off")
|
| 49 |
|
|
|
|
| 24 |
fdic = {
|
| 25 |
"family" : "Impact",
|
| 26 |
"style" : "italic",
|
| 27 |
+
"size" : 1,
|
| 28 |
"color" : "yellow",
|
| 29 |
"weight" : "bold"
|
| 30 |
}
|
|
|
|
| 43 |
w, h = prediction['box']['xmax'] - prediction['box']['xmin'], prediction['box']['ymax'] - prediction['box']['ymin']
|
| 44 |
|
| 45 |
ax.add_patch(plt.Rectangle((x, y), w, h, fill=False, color=selected_color, linewidth=3))
|
| 46 |
+
ax.text(x, y, f"{prediction['label']}: {round(prediction['score']*100, 1)}%", fontdict=fdic)
|
| 47 |
|
| 48 |
plt.axis("off")
|
| 49 |
|