mishtert commited on
Commit
dc389ab
·
1 Parent(s): 9d3b7c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ from PIL import Image, ImageDraw, ImageFont
18
 
19
  processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv2-base-uncased")
20
  #model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-finetuned-funsd")
21
- model = LayoutLMv2ForTokenClassification.from_pretrained("mishtert/Invoice_extraction_categorization")
22
 
23
  # load image example
24
  #dataset = load_dataset("nielsr/funsd", split="test")
 
18
 
19
  processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv2-base-uncased")
20
  #model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-finetuned-funsd")
21
+ model = LayoutLMv2ForTokenClassification.from_pretrained("mishtert/iec")
22
 
23
  # load image example
24
  #dataset = load_dataset("nielsr/funsd", split="test")