Spaces:
Running
Running
Update sroie_inference.py
Browse files- sroie_inference.py +1 -1
sroie_inference.py
CHANGED
@@ -97,7 +97,7 @@ def prediction(image):
|
|
97 |
k = k[2:]
|
98 |
|
99 |
blur_boxes = []
|
100 |
-
for prediction, box in zip(
|
101 |
if prediction != 'O' and prediction[2:] != 'TOTAL':
|
102 |
blur_boxes.append(box)
|
103 |
|
|
|
97 |
k = k[2:]
|
98 |
|
99 |
blur_boxes = []
|
100 |
+
for prediction, box in zip(true_predictions, true_boxes):
|
101 |
if prediction != 'O' and prediction[2:] != 'TOTAL':
|
102 |
blur_boxes.append(box)
|
103 |
|