Spaces:

mp-02
/

LayoutLMv3_for_recepits

Sleeping

mp-02 commited on Aug 9, 2024

Commit

05e8dab

verified ·

1 Parent(s): 0a411ad

Update cord_inference.py

Files changed (1) hide show

cord_inference.py CHANGED Viewed

@@ -5,7 +5,7 @@ from PIL import Image, ImageDraw, ImageFont
 from utils import OCR, unnormalize_box
-id2label: {
     "0": "O",
     "1": "B-MENU.CNT",
     "2": "B-MENU.DISCOUNTPRICE",
@@ -47,7 +47,7 @@ id2label: {
     "38": "I-TOTAL.TOTAL_PRICE"
   }
-label2id: {
     "B-MENU.CNT": 1,
     "B-MENU.DISCOUNTPRICE": 2,
     "B-MENU.NM": 3,

 from utils import OCR, unnormalize_box
+id2label = {
     "0": "O",
     "1": "B-MENU.CNT",
     "2": "B-MENU.DISCOUNTPRICE",
     "38": "I-TOTAL.TOTAL_PRICE"
   }
+label2id = {
     "B-MENU.CNT": 1,
     "B-MENU.DISCOUNTPRICE": 2,
     "B-MENU.NM": 3,