gwkrsrch commited on
Commit
bf229cc
1 Parent(s): 8d1eb5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
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(