Spaces:
Running
Running
Update sroie_inference.py
Browse files- sroie_inference.py +1 -1
sroie_inference.py
CHANGED
|
@@ -86,7 +86,7 @@ def prediction(image):
|
|
| 86 |
d = {k: v.strip() for (k, v) in d.items()}
|
| 87 |
|
| 88 |
if "O" in d: d.pop("O")
|
| 89 |
-
if "
|
| 90 |
|
| 91 |
blur_boxes = []
|
| 92 |
for prediction, box in zip(true_predictions, true_boxes):
|
|
|
|
| 86 |
d = {k: v.strip() for (k, v) in d.items()}
|
| 87 |
|
| 88 |
if "O" in d: d.pop("O")
|
| 89 |
+
if "TOTAL" in d: d.pop("TOTAL")
|
| 90 |
|
| 91 |
blur_boxes = []
|
| 92 |
for prediction, box in zip(true_predictions, true_boxes):
|