Spaces:

mp-02
/

LayoutLMv3_for_recepits

Running

mp-02 commited on 5 days ago

Commit

909061c

•

1 Parent(s): f4b0061

Update sroie_inference.py

Files changed (1) hide show

sroie_inference.py CHANGED Viewed

@@ -90,7 +90,7 @@ def prediction(image):
     blur_boxes = []
     for prediction, box in zip(true_predictions, true_boxes):
-        if prediction != 'O' and prediction != 'TOTAL':
             blur_boxes.append(box)
     image = (blur(image, blur_boxes))

     blur_boxes = []
     for prediction, box in zip(true_predictions, true_boxes):
+        if prediction != 'O' and prediction != 'S-TOTAL':
             blur_boxes.append(box)
     image = (blur(image, blur_boxes))