jinhybr commited on
Commit
cc8b275
1 Parent(s): d0f0aa4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ from transformers import LayoutLMv2Processor, LayoutLMv2ForTokenClassification
16
  from datasets import load_dataset
17
  from PIL import Image, ImageDraw, ImageFont
18
 
19
- processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv2-base-uncased")
20
- model = LayoutLMv2ForTokenClassification.from_pretrained("jinhybr/OCR-LM-v1")
21
 
22
  # load image example
23
  dataset = load_dataset("nielsr/funsd", split="test")
 
16
  from datasets import load_dataset
17
  from PIL import Image, ImageDraw, ImageFont
18
 
19
+ processor = LayoutLMv2Processor.from_pretrained("jinhybr/OCR-LM-v1")
20
+ model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-finetuned-funsd")
21
 
22
  # load image example
23
  dataset = load_dataset("nielsr/funsd", split="test")