Spaces:
Runtime error
Runtime error
sshi
commited on
Commit
•
4a0733d
1
Parent(s):
224b9f6
App bug fix.
Browse files
app.py
CHANGED
|
@@ -107,7 +107,7 @@ def plot_results(pil_img, prob, boxes):
|
|
| 107 |
|
| 108 |
for p, (xmin, ymin, xmax, ymax) in zip(prob, boxes.tolist()):
|
| 109 |
cl = p.argmax()
|
| 110 |
-
c = colors[cl]
|
| 111 |
|
| 112 |
draw.rectangle([xmin, ymin, xmax - xmin, ymax - ymin], outline=c, width=2)
|
| 113 |
draw.text(
|
|
|
|
| 107 |
|
| 108 |
for p, (xmin, ymin, xmax, ymax) in zip(prob, boxes.tolist()):
|
| 109 |
cl = p.argmax()
|
| 110 |
+
c = tuple(colors[cl])
|
| 111 |
|
| 112 |
draw.rectangle([xmin, ymin, xmax - xmin, ymax - ymin], outline=c, width=2)
|
| 113 |
draw.text(
|