Upload app.py

app.py

@@ -22,8 +22,7 @@ model = LayoutLMv2ForTokenClassification.from_pretrained("amir22010/layoutxlm-xf
 
 # load image example
 dataset = load_dataset("ranpox/xfund", 'xfund.ja', split="validation")
-print(dataset[0])
-image = Image.open(dataset[0]["image"]).convert("RGB")
+image = Image.open(dataset[0]["image"][0]).convert("RGB")
 image = Image.open("./invoice.png")
 image.save("document.png")
 # define id2label, label2color