binery commited on
Commit
d4e2d26
1 Parent(s): 5b2fe1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ image.save("cord_sample_receipt1.png")
20
  image = Image.open("./sample_image_cord_test_receipt_00012.png")
21
  image.save("cord_sample_receipt2.png")
22
 
23
- pretrained_model = DonutModel.from_pretrained("Raj-Master/donut-demo-123/tree/main")
24
  pretrained_model.encoder.to(torch.bfloat16)
25
  pretrained_model.eval()
26
 
 
20
  image = Image.open("./sample_image_cord_test_receipt_00012.png")
21
  image.save("cord_sample_receipt2.png")
22
 
23
+ pretrained_model = DonutModel.from_pretrained("Raj-Master/donut-demo-123/blob/main")
24
  pretrained_model.encoder.to(torch.bfloat16)
25
  pretrained_model.eval()
26