Geewook Kim
commited on
Commit
·
ed5b9bd
1
Parent(s):
4aa30ec
Update app.py
Browse files
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("
|
|
|
|
|
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=[["
|
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()
|