Spaces:

mp-02
/

LayoutLMv3_for_recepits

Running

mp-02 commited on 6 days ago

Commit

b3d156d

•

1 Parent(s): e3e9c8c

Update sroie_inference.py

Files changed (1) hide show

sroie_inference.py CHANGED Viewed

@@ -6,9 +6,9 @@ from transformers import LayoutLMv3TokenizerFast, LayoutLMv3Processor, LayoutLMv
 from utils import OCR, unnormalize_box
-tokenizer = LayoutLMv3TokenizerFast.from_pretrained("mp-02/layoutlmv3-finetuned-sroie", apply_ocr=False)
-processor = LayoutLMv3Processor.from_pretrained("mp-02/layoutlmv3-finetuned-sroie", apply_ocr=False)
-model = LayoutLMv3ForTokenClassification.from_pretrained("mp-02/layoutlmv3-finetuned-sroie")
 id2label = model.config.id2label
 label2id = model.config.label2id

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