Upload app.py

app.py

@@ -21,7 +21,7 @@ processor = LayoutXLMProcessor.from_pretrained("amir22010/layoutxlm-xfund-ja")
 model = LayoutLMv2ForTokenClassification.from_pretrained("amir22010/layoutxlm-xfund-ja",num_labels = 7)
 
 # load image example
-dataset = load_dataset("ranpox/xfund", 'xfund.ja', split="test")
+dataset = load_dataset("ranpox/xfund", 'xfund.ja', split="validation")
 image = Image.open(dataset[0]["image_path"]).convert("RGB")
 image = Image.open("./invoice.png")
 image.save("document.png")