Update app.py

app.py

@@ -17,12 +17,12 @@ 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/Invoice_extraction_categorization")
+#model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-finetuned-funsd")
+model = LayoutLMv2ForTokenClassification.from_pretrained("Mishtert/Invoice_extraction_categorization")
 
 # load image example
-dataset = load_dataset("nielsr/funsd", split="test")
-#dataset = load_dataset("Mishtert/niefunsd", split="test")
+#dataset = load_dataset("nielsr/funsd", split="test")
+dataset = load_dataset("Mishtert/niefunsd", split="test")
 image = Image.open(dataset[0]["image_path"]).convert("RGB")
 image = Image.open("./invoice.png")
 image.save("document.png")