Spaces:

imflash217
/

doc_ai_kmbs

Runtime error

imflash217 commited on Aug 4, 2022

Commit

a7f213e

•

1 Parent(s): f2b5b7d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,8 +16,8 @@ from transformers import LayoutLMv2Processor, LayoutLMv2ForTokenClassification
 from datasets import load_dataset
 from PIL import Image, ImageDraw, ImageFont
-processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutxlm-base")
-model = LayoutLMv2ForTokenClassification.from_pretrained("microsoft/layoutxlm-base")
 # load image example
 dataset = load_dataset("nielsr/funsd", split="test")

 from datasets import load_dataset
 from PIL import Image, ImageDraw, ImageFont
+processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv2-base-uncased")
+model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-finetuned-funsd")
 # load image example
 dataset = load_dataset("nielsr/funsd", split="test")