Update app.py

app.py

@@ -18,11 +18,11 @@ 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("mishtert/Invoice_extraction_categorization")
 
 # load image example
 #dataset = load_dataset("nielsr/funsd", split="test")
-dataset = load_dataset("Mishtert/niefunsd", 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")