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 |
|