Update app.py

app.py

@@ -29,7 +29,7 @@ model = AutoModelForTokenClassification.from_pretrained("jinhybr/OCR-LayoutLMv3"
 
 # load image example
 dataset = load_dataset("nielsr/funsd-layoutlmv3", split="test")
-image = Image.open(dataset[0]["image_path"]).convert("RGB")
+image = Image.open(dataset[0]).convert("RGB")
 image = Image.open("./example_lm3.png")
 image.save("document.jpg")
 # define id2label, label2color
@@ -37,6 +37,20 @@ labels = dataset.features['ner_tags'].feature.names
 id2label = {v: k for v, k in enumerate(labels)}
 label2color = {'question':'blue', 'answer':'green', 'header':'orange', 'other':'violet'}
 
+
+
+#####
+
+
+
+
+
+
+
+
+
+
+
 def unnormalize_box(bbox, width, height):
      return [
          width * (bbox[0] / 1000),