Ammar-alhaj-ali commited on
Commit
fd15a12
1 Parent(s): 934c0be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ model = LayoutLMv3ForTokenClassification.from_pretrained("Ammar-alhaj-ali/Layout
20
  # load image example
21
  dataset = load_dataset("nielsr/funsd", split="test")
22
  image = Image.open(dataset[0]["image_path"]).convert("RGB")
23
- image = Image.open("./invoice.png")
24
  image.save("document.png")
25
  # define id2label, label2color
26
  labels = dataset.features['ner_tags'].feature.names
 
20
  # load image example
21
  dataset = load_dataset("nielsr/funsd", split="test")
22
  image = Image.open(dataset[0]["image_path"]).convert("RGB")
23
+ #image = Image.open("./invoice.png")
24
  image.save("document.png")
25
  # define id2label, label2color
26
  labels = dataset.features['ner_tags'].feature.names