App push

app.py

@@ -15,9 +15,9 @@ model = LayoutLMv2ForTokenClassification.from_pretrained("Theivaprakasham/layout
 
 # load image example
 dataset = load_dataset("darentang/generated", split="test")
-Image.open(dataset[50]["image_path"]).convert("RGB").save("example1.png")
-Image.open(dataset[14]["image_path"]).convert("RGB").save("example2.png")
-Image.open(dataset[20]["image_path"]).convert("RGB").save("example3.png")
+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")
 # define id2label, label2color
 labels = dataset.features['ner_tags'].feature.names
 id2label = {v: k for v, k in enumerate(labels)}