Update app.py
Browse files
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/
|
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")
|