Spaces:
Build error
Build error
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -63,7 +63,7 @@ def prediction(image):
|
|
| 63 |
else:
|
| 64 |
d[i] = d[i] + ", " + true_words[id]
|
| 65 |
|
| 66 |
-
|
| 67 |
|
| 68 |
if "O" in d: d.pop("O")
|
| 69 |
|
|
|
|
| 63 |
else:
|
| 64 |
d[i] = d[i] + ", " + true_words[id]
|
| 65 |
|
| 66 |
+
d = {k: v.strip() for (k, v) in d.items()}
|
| 67 |
|
| 68 |
if "O" in d: d.pop("O")
|
| 69 |
|