Update app.py

app.py

@@ -17,8 +17,8 @@ from datasets import load_dataset
 from PIL import Image, ImageDraw, ImageFont
 
 processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv2-base-uncased")
-#model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-finetuned-funsd")
-model = LayoutLMv2ForTokenClassification.from_pretrained("mishtert/iec")
+model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-finetuned-funsd")
+#model = LayoutLMv2ForTokenClassification.from_pretrained("mishtert/iec")
 
 # load image example
 #dataset = load_dataset("nielsr/funsd", split="test")
@@ -81,8 +81,8 @@ def process_image(image):
 title = "Interactive demo: Invoice Extraction & Categorization"
 description = "Text extracted and annotated QUESTION/ANSWER/HEADER/OTHER." 
 
-examples =[['document.png']]
-#examples =[['document.png'],['invoice2.png'],['invoice3.png']]
+#examples =[['document.png']]
+examples =[['document.png'],['invoice2.png'],['invoice3.png']]
 
 css = ".output-image, .input-image {height: 40rem !important; width: 100% !important;}"
 #css = "@media screen and (max-width: 600px) { .output_image, .input_image {height:20rem !important; width: 100% !important;} }"