Update sroie_inference.py

sroie_inference.py

@@ -86,11 +86,11 @@ def prediction(image):
     d = {k: v.strip() for (k, v) in d.items()}
 
     if "O" in d: d.pop("O")
-    if "TOTAL" in d: d.pop("TOTAL")
+    if "S-TOTAL" in d: d.pop("S-TOTAL")
         
     blur_boxes = []
     for prediction, box in zip(true_predictions, true_boxes):
-        if prediction != 'O' and prediction != 'TOTAL':
+        if prediction != 'O' and prediction != 'S-TOTAL':
             blur_boxes.append(box)
 
     image = (blur(image, blur_boxes))