Update app.py

app.py

@@ -4,8 +4,8 @@ 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")
+processor = DonutProcessor.from_pretrained("./donut-base-finetuned-inv")
+model = VisionEncoderDecoderModel.from_pretrained("./donut-base-finetuned-inv")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
@@ -39,7 +39,7 @@ def process_document(image):
     
     return processor.token2json(sequence)
 
-description = "Gradio Demo for Donut, an instance of `VisionEncoderDecoderModel` fine-tuned on CORD (document parsing). 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."
+description = "Gradio Demo for Donut, an instance of `VisionEncoderDecoderModel` fine-tuned on Invoices. 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>"
 
 demo = gr.Interface(