Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
|
|
20 |
model = LayoutLMv3ForTokenClassification.from_pretrained("nielsr/layoutlmv3-finetuned-cord")
|
21 |
|
22 |
# load image example
|
23 |
-
dataset = load_dataset("nielsr/cord-layoutlmv3", split="test")
|
24 |
#image = Image.open(dataset[0]["image_path"]).convert("RGB")
|
25 |
image = Image.open("./test0.jpeg")
|
26 |
# define id2label, label2color
|
|
|
20 |
model = LayoutLMv3ForTokenClassification.from_pretrained("nielsr/layoutlmv3-finetuned-cord")
|
21 |
|
22 |
# load image example
|
23 |
+
dataset = load_dataset("nielsr/cord-layoutlmv3", split="test", trust_remote_code=True)
|
24 |
#image = Image.open(dataset[0]["image_path"]).convert("RGB")
|
25 |
image = Image.open("./test0.jpeg")
|
26 |
# define id2label, label2color
|