amir22010 commited on
Commit
24240a3
·
1 Parent(s): 998356d

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ processor = LayoutXLMProcessor.from_pretrained("amir22010/layoutxlm-xfund-ja")
21
  model = LayoutLMv2ForTokenClassification.from_pretrained("amir22010/layoutxlm-xfund-ja",num_labels = 7)
22
 
23
  # load image example
24
- dataset = load_dataset("ranpox/xfund", 'xfun.ja', split="test")
25
  image = Image.open(dataset[0]["image_path"]).convert("RGB")
26
  image = Image.open("./invoice.png")
27
  image.save("document.png")
 
21
  model = LayoutLMv2ForTokenClassification.from_pretrained("amir22010/layoutxlm-xfund-ja",num_labels = 7)
22
 
23
  # load image example
24
+ dataset = load_dataset("ranpox/xfund", 'xfund.ja', split="test")
25
  image = Image.open(dataset[0]["image_path"]).convert("RGB")
26
  image = Image.open("./invoice.png")
27
  image.save("document.png")