Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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("
|
20 |
-
model = LayoutLMv2ForTokenClassification.from_pretrained("
|
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")
|