Update app.py

app.py

@@ -4,8 +4,8 @@ import torch
 from PIL import Image
 from transformers import DonutProcessor, VisionEncoderDecoderModel
 
-task_prompt = f"<s_cord-v2>"
-pretrained_path = "Chan-yeong/donut-cord-sample-demo"
+task_prompt = f"<s_sroie>"
+pretrained_path = "Chan-yeong/donut-sroie-company-sample-demo"
 
 processor = DonutProcessor.from_pretrained(pretrained_path)
 pretrained_model = VisionEncoderDecoderModel.from_pretrained(pretrained_path)