Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,6 @@ 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)
|
29 |
pretrained_model.eval()
|
30 |
|
31 |
demo = gr.Interface(
|
|
|
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.eval()
|
29 |
|
30 |
demo = gr.Interface(
|