rajistics commited on
Commit
dc13441
β€’
1 Parent(s): 7179a4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
20
  model = LayoutLMv3ForTokenClassification.from_pretrained("nielsr/layoutlmv3-finetuned-cord")
21
 
22
  # load image example
23
- dataset = load_dataset("nielsr/cord-layoutlmv3", split="test")
24
  #image = Image.open(dataset[0]["image_path"]).convert("RGB")
25
  image = Image.open("./test0.jpeg")
26
  # define id2label, label2color
 
20
  model = LayoutLMv3ForTokenClassification.from_pretrained("nielsr/layoutlmv3-finetuned-cord")
21
 
22
  # load image example
23
+ dataset = load_dataset("nielsr/cord-layoutlmv3", split="test", trust_remote_code=True)
24
  #image = Image.open(dataset[0]["image_path"]).convert("RGB")
25
  image = Image.open("./test0.jpeg")
26
  # define id2label, label2color