Update app.py

app.py

@@ -7,6 +7,7 @@ https://github.com/clovaai/donut
 import gradio as gr
 import torch
 from PIL import Image
+from transformers import DonutProcessor, VisionEncoderDecoderModel
 
 from donut import DonutModel
 
@@ -19,9 +20,12 @@ def demo_process(input_img):
 task_prompt = f"<s_cord-v2>"
 
 
-pretrained_model = DonutModel.from_pretrained("to-be/ID_cards_v1",revision="main")
+#pretrained_model = DonutModel.from_pretrained("to-be/ID_cards_v1",revision="main")
+pretrained_model = VisionEncoderDecoderModel.from_pretrained("to-be/ID_cards_v1")
 pretrained_model.eval()
 
+
+
 demo = gr.Interface(
     fn=demo_process,
     inputs= gr.inputs.Image(type="pil"),