Spaces:
Runtime error
Runtime error
fix: id of donut model inside unstructuredio
Browse files
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("
|
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 |
|