mp-02 commited on
Commit
fcd9252
·
verified ·
1 Parent(s): f0bd0ec

Update cord_inference.py

Browse files
Files changed (1) hide show
  1. cord_inference.py +1 -1
cord_inference.py CHANGED
@@ -51,7 +51,7 @@ label2id = {v: k for k, v in enumerate(label_list)}
51
 
52
  tokenizer = LayoutLMv3TokenizerFast.from_pretrained("mp-02/layoutlmv3-finetuned-cord", apply_ocr=False)
53
  processor = LayoutLMv3Processor.from_pretrained("mp-02/layoutlmv3-finetuned-cord", apply_ocr=False)
54
- model = LayoutLMv3ForTokenClassification.from_pretrained("mp-02/layoutlmv3-finetuned-cord")
55
 
56
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
57
  model.to(device)
 
51
 
52
  tokenizer = LayoutLMv3TokenizerFast.from_pretrained("mp-02/layoutlmv3-finetuned-cord", apply_ocr=False)
53
  processor = LayoutLMv3Processor.from_pretrained("mp-02/layoutlmv3-finetuned-cord", apply_ocr=False)
54
+ model = LayoutLMv3ForTokenClassification.from_pretrained("mp-02/layoutlmv3-finetuned-cord", ignore_mismatched_sizes=True)
55
 
56
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
57
  model.to(device)