gwkrsrch commited on
Commit
ed5b9bd
1 Parent(s): 4aa30ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -20,7 +20,9 @@ def demo_process(input_img):
20
  task_prompt = f"<s_cord-v2>"
21
 
22
  image = Image.open("./sample_image_cord_test_receipt_00004.png")
23
- image.save("cord_sample_receipt.png")
 
 
24
 
25
  pretrained_model = DonutModel.from_pretrained("naver-clova-ix/donut-base-finetuned-cord-v2")
26
  pretrained_model.encoder.to(torch.bfloat16)
@@ -38,7 +40,7 @@ More CORD receipt images are available at https://huggingface.co/datasets/naver-
38
  More details are available at:
39
  - Paper: https://arxiv.org/abs/2111.15664
40
  - GitHub: https://github.com/clovaai/donut""",
41
- examples=[["cord_sample_receipt.png"]],
42
  cache_examples=False,
43
  )
44
  demo.launch()
 
20
  task_prompt = f"<s_cord-v2>"
21
 
22
  image = Image.open("./sample_image_cord_test_receipt_00004.png")
23
+ image.save("cord_sample_receipt1.png")
24
+ image = Image.open("./sample_image_cord_test_receipt_00012.png")
25
+ image.save("cord_sample_receipt2.png")
26
 
27
  pretrained_model = DonutModel.from_pretrained("naver-clova-ix/donut-base-finetuned-cord-v2")
28
  pretrained_model.encoder.to(torch.bfloat16)
 
40
  More details are available at:
41
  - Paper: https://arxiv.org/abs/2111.15664
42
  - GitHub: https://github.com/clovaai/donut""",
43
+ examples=[["cord_sample_receipt1.png"], ["cord_sample_receipt2.png"]],
44
  cache_examples=False,
45
  )
46
  demo.launch()