Update app.py

app.py

@@ -21,8 +21,8 @@ model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-fine
 #model = LayoutLMv2ForTokenClassification.from_pretrained("mishtert/iec")
 
 # 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")