image loading

app.py

@@ -20,10 +20,10 @@ processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
 model = LayoutLMv3ForTokenClassification.from_pretrained("nielsr/layoutlmv3-finetuned-cord")
 
 # load image example
-dataset = load_dataset("nielsr/cord-layoutlmv3", split="test")
-image = Image.open(dataset[0]["image_path"]).convert("RGB")
-#image = Image.open("./test0.jpg")
-image.save("document.png")
+#dataset = load_dataset("nielsr/cord-layoutlmv3", split="test")
+#image = Image.open(dataset[0]["image_path"]).convert("RGB")
+image = Image.open("./test0.jpeg")
+#image.save("document.png")
 # define id2label, label2color
 labels = dataset.features['ner_tags'].feature.names
 id2label = {v: k for v, k in enumerate(labels)}