Update app.py

app.py

@@ -29,9 +29,9 @@ model = AutoModelForTokenClassification.from_pretrained("jinhybr/OCR-LayoutLMv3"
 
 # load image example
 dataset = load_dataset("nielsr/funsd-layoutlmv3", split="test")
-image = Image.open(dataset[0]).convert("RGB")
-image = Image.open("./example_lm3.png")
-image.save("document.jpg")
+Image.open(dataset[0]["image_path"]).convert("RGB").save("example_lm3.png")
+
+
 # define id2label, label2color
 labels = dataset.features['ner_tags'].feature.names
 id2label = {v: k for v, k in enumerate(labels)}