Upload 2 files

app.py

@@ -21,10 +21,10 @@ 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="validation")
-image = Image.open(dataset[0]["image"][0]).convert("RGB")
-image = Image.open("./invoice.png")
-image.save("document.png")
+#dataset = load_dataset("ranpox/xfund", 'xfund.ja', split="validation")
+#image = Image.open(dataset[0]["image"][0]).convert("RGB")
+image = Image.open("./ja_val_49.jpg").convert("RGB")
+image.save("document.jpg")
 # define id2label, label2color
 labels = [
     'O',