Spaces:

mp-02
/

LayoutLMv3_for_recepits

Sleeping

mp-02 commited on Nov 14, 2024

Commit

9b31f19

•

1 Parent(s): 129680c

Update cord_inference.py

Files changed (1) hide show

cord_inference.py CHANGED Viewed

@@ -5,9 +5,9 @@ from PIL import Image, ImageDraw, ImageFont
 from utils import OCR, unnormalize_box
-tokenizer = LayoutLMv3TokenizerFast.from_pretrained("mp-02/layoutlmv3-large-cord", apply_ocr=False)
-processor = LayoutLMv3Processor.from_pretrained("mp-02/layoutlmv3-large-cord", apply_ocr=False)
-model = LayoutLMv3ForTokenClassification.from_pretrained("mp-02/layoutlmv3-large-cord")
 id2label = model.config.id2label
 label2id = model.config.label2id

 from utils import OCR, unnormalize_box
+tokenizer = LayoutLMv3TokenizerFast.from_pretrained("mp-02/layoutlmv3-base-cord", apply_ocr=False)
+processor = LayoutLMv3Processor.from_pretrained("mp-02/layoutlmv3-base-cord", apply_ocr=False)
+model = LayoutLMv3ForTokenClassification.from_pretrained("mp-02/layoutlmv3-base-cord")
 id2label = model.config.id2label
 label2id = model.config.label2id