gwkrsrch commited on
Commit
3c90045
β€’
1 Parent(s): 7fde610

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,8 +19,8 @@ def demo_process(input_img):
19
 
20
  task_prompt = f"<s_cord-v2>"
21
 
22
- image = Image.open("./sample_image_cord_test_receipt_00004.jpg")
23
- image.save("cord_sample_receipt.jpg")
24
 
25
  pretrained_model = DonutModel.from_pretrained("naver-clova-ix/donut-base-finetuned-cord-v2")
26
  pretrained_model.encoder.to(torch.bfloat16)
@@ -32,7 +32,7 @@ demo = gr.Interface(
32
  outputs="json",
33
  title=f"Donut 🍩 demonstration for `cord-v2` task",
34
  description="This is a Donut 🍩 demonstration for `cord-v2` task.\nFor more receipt sample images: https://huggingface.co/datasets/naver-clova-ix/cord-v2 \nPaper: https://arxiv.org/abs/2111.15664 \nGitHub: https://github.com/clovaai/donut",
35
- examples=[["cord_sample_receipt.jpg"]],
36
  cache_examples=False,
37
  )
38
  demo.launch()
 
19
 
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)
 
32
  outputs="json",
33
  title=f"Donut 🍩 demonstration for `cord-v2` task",
34
  description="This is a Donut 🍩 demonstration for `cord-v2` task.\nFor more receipt sample images: https://huggingface.co/datasets/naver-clova-ix/cord-v2 \nPaper: https://arxiv.org/abs/2111.15664 \nGitHub: https://github.com/clovaai/donut",
35
+ examples=[["cord_sample_receipt.png"]],
36
  cache_examples=False,
37
  )
38
  demo.launch()