Update app.py

app.py

@@ -26,9 +26,9 @@ model = AutoModelForTokenClassification.from_pretrained("SickBoy/layoutlm_docume
 
 # load image example
 dataset = load_dataset("SickBoy/layout_documents", split="train")
-Image.open(dataset[2]["image_path"]).convert("RGB").save("example1.png")
-Image.open(dataset[1]["image_path"]).convert("RGB").save("example2.png")
-Image.open(dataset[0]["image_path"]).convert("RGB").save("example3.png")
+dataset[2]["image"].convert("RGB").save("example1.png")
+dataset[1]["image"].convert("RGB").save("example2.png")
+dataset[0]["image"].convert("RGB").save("example3.png")
 # define id2label, label2color
 #labels = dataset.features['ner_tags'].feature.names
 labels = ['O', 'HEADER', 'SUBHEADER', 'TEXTO', 'NUMERAL', 'RESUMEN']