Update app.py

app.py

@@ -26,10 +26,10 @@ processor = AutoProcessor.from_pretrained("Ammar-alhaj-ali/LayoutLMv3-Fine-Tunin
 model = AutoModelForTokenClassification.from_pretrained("Ammar-alhaj-ali/LayoutLMv3-Fine-Tuning-FUNSD")
 
 # load image example
-dataset = load_dataset("darentang/generated", split="test")
-Image.open(dataset[2]["image_path"]).convert("RGB").save("img1.png")
-Image.open(dataset[1]["image_path"]).convert("RGB").save("img2.png")
-Image.open(dataset[0]["image_path"]).convert("RGB").save("img3.png")
+#dataset = load_dataset("nielsr/funsd-layoutlmv3", split="test")
+#Image.open(dataset[2]["image_path"]).convert("RGB").save("img1.png")
+#Image.open(dataset[1]["image_path"]).convert("RGB").save("img2.png")
+#Image.open(dataset[0]["image_path"]).convert("RGB").save("img3.png")
 # define id2label, label2color
 labels = ['O', 'B-HEADER', 'I-HEADER', 'B-QUESTION', 'I-QUESTION', 'B-ANSWER', 'I-ANSWER']
 id2label = {v: k for v, k in enumerate(labels)}