laverdes commited on
Commit
78588fe
·
1 Parent(s): 24691cd

fix: id of donut model inside unstructuredio

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ image = Image.open(f"./img/receipt-{receipt}.jpg")
30
  st.image(image, caption='Your target receipt')
31
 
32
  st.text(f'baking the 🍩...')
33
- pretrained_model = VisionEncoderDecoderModel.from_pretrained("unstructured/donut-base-sroie")
34
  pretrained_model.encoder.to(torch.bfloat16)
35
  pretrained_model.eval()
36
 
 
30
  st.image(image, caption='Your target receipt')
31
 
32
  st.text(f'baking the 🍩...')
33
+ pretrained_model = VisionEncoderDecoderModel.from_pretrained("unstructuredio/donut-base-sroie")
34
  pretrained_model.encoder.to(torch.bfloat16)
35
  pretrained_model.eval()
36