fix: add device and sent model to device

app.py

@@ -69,6 +69,10 @@ st.image(image, caption='Your target receipt')
 st.text(f'baking the 🍩...')
 processor = DonutProcessor.from_pretrained("unstructuredio/donut-base-sroie")
 pretrained_model = VisionEncoderDecoderModel.from_pretrained("unstructuredio/donut-base-sroie")
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pretrained_model.to(device)
+
 pretrained_model.encoder.to(torch.bfloat16)
 pretrained_model.eval()