Geewook Kim
commited on
Commit
·
3c90045
1
Parent(s):
7fde610
Update app.py
Browse files
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.
|
23 |
-
image.save("cord_sample_receipt.
|
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.
|
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()
|