Update app.py

app.py

@@ -16,7 +16,7 @@ model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-fine
 
 # load image example
 dataset = load_dataset("nielsr/funsd", split="test", trust_remote_code=True)
-image = Image.open(dataset[0]["image_path"]).convert("RGB")
+image = Image.open(dataset[0]["image"]).convert("RGB")
 image = Image.open("./invoice.png")
 image.save("document.png")
 # define id2label, label2color