Update app.py

app.py

@@ -3,9 +3,9 @@ import gradio as gr
 import torch
 
 from transformers import DonutProcessor, VisionEncoderDecoderModel
-
-processor = DonutProcessor.from_pretrained("naver-clova-ix/donut-base-finetuned-cord-v2")
-model = VisionEncoderDecoderModel.from_pretrained("naver-clova-ix/donut-base-finetuned-cord-v2")
+#naver-clova-ix/donut-base-finetuned-cord-v2
+processor = DonutProcessor.from_pretrained("jinhybr/OCR-Donut-CORD")
+model = VisionEncoderDecoderModel.from_pretrained("jinhybr/OCR-Donut-CORD")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
@@ -39,8 +39,8 @@ def process_document(image):
     
     return processor.token2json(sequence)
 
-description = "Gradio Demo for Donut, an instance of `VisionEncoderDecoderModel` fine-tuned on SROIE dataset. To use it, simply upload your image and click 'submit', or click one of the examples to load them. Read more at the links below."
-article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.15664' target='_blank'>Donut: OCR-free Document Understanding Transformer</a> | <a href='https://github.com/clovaai/donut' target='_blank'>Github Repo</a></p>"
+description = "Gradio Demo for Donut, an instance of `VisionEncoderDecoderModel` fine-tuned on CORD dataset. To use it, simply upload your image and click 'submit', or click one of the examples to load them. Read more at the links below."
+
 
 demo = gr.Interface(
     fn=process_document,