Update app.py

app.py

@@ -43,8 +43,8 @@ model = AutoModelForTokenClassification.from_pretrained("jinhybr/OCR-LayoutLMv3-
 
 # load image example
 dataset = load_dataset("jinhybr/WildReceipt", split="test")
-Image.open(dataset[18]["image_path"]).convert("RGB").save("example1.png")
-Image.open(dataset[19]["image_path"]).convert("RGB").save("example2.png")
+Image.open(dataset[1]["image_path"]).convert("RGB").save("example1.png")
+Image.open(dataset[3]["image_path"]).convert("RGB").save("example2.png")
 Image.open(dataset[25]["image_path"]).convert("RGB").save("example3.png")
 # define id2label, label2color
 labels = dataset.features['ner_tags'].feature.names