Update app.py

app.py

@@ -52,29 +52,29 @@ processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base",appl
 model = LayoutLMv3ForTokenClassification.from_pretrained("nielsr/layoutlmv3-finetuned-funsd")
 
 dataset = load_dataset("nielsr/funsd", split="test")
-image = Image.open(dataset[0]["image_path"]).convert("RGB")
-image = Image.open("./invoice.png")
-image.save("document1.png")
+#image = Image.open(dataset[0]["image_path"]).convert("RGB")
+#image = Image.open("./invoice.png")
+#image.save("document1.png")
 
-image = Image.open(dataset[1]["image_path"]).convert("RGB")
-image = Image.open("./invoice2.png")
-image.save("document2.png")
+#image = Image.open(dataset[1]["image_path"]).convert("RGB")
+#image = Image.open("./invoice2.png")
+#image.save("document2.png")
 
-image = Image.open(dataset[2]["image_path"]).convert("RGB")
-image = Image.open("./invoice3.png")
-image.save("document3.png")
+#image = Image.open(dataset[2]["image_path"]).convert("RGB")
+#image = Image.open("./invoice3.png")
+#image.save("document3.png")
 
 
 #dataset = load_dataset("nielsr/funsd-layoutlmv3")
 
-#example = dataset["test"][0]
-#example["image"].save("example1.png")
+example = dataset["test"][0]
+example["image"].save("example1.png")
 
-#example1 = dataset["test"][1]
-#example1["image"].save("example2.png")
+example1 = dataset["test"][1]
+example1["image"].save("example2.png")
 
-#example2 = dataset["test"][2]
-#example2["image"].save("example3.png")
+example2 = dataset["test"][2]
+example2["image"].save("example3.png")
 
 #example2["image"]
 
@@ -151,7 +151,7 @@ description = "Extraction of Form or Invoice Extraction - We use Microsoft's Lay
 
 article="<b>References</b><br>[1] Y. Xu et al., “LayoutLMv3: Pre-training for Document AI with Unified Text and Image Masking.” 2022. <a href='https://arxiv.org/abs/2204.08387'>Paper Link</a><br>[2]  <a href='https://github.com/NielsRogge/Transformers-Tutorials/tree/master/LayoutLMv3'>LayoutLMv3 training and inference</a>" 
 
-examples =[['document1.png'],['document1.png'],['document1.png']]
+examples =[['example1.png'],['example1.png'],['example1.png']]
 
 css = """.output_image, .input_image {height: 600px !important}"""