Update app.py

app.py

@@ -10,9 +10,9 @@ from PIL import Image
 
 from donut import DonutModel
 from transformers import DonutProcessor, VisionEncoderDecoderModel
-from huggingface_hub import notebook_login
+#from huggingface_hub import notebook_login
 
-huggingface_hub.notebook_login(token=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)
+#huggingface_hub.notebook_login(token=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)
 
 def demo_process(input_img):
     global pretrained_model, task_prompt, task_name
@@ -31,7 +31,8 @@ image.save("csroie_test1.jpg")
 image = Image.open("./001.jpg")
 image.save("sroie_test2.jpg")
 
-pretrained_model = DonutModel.from_pretrained("sav7669/donut-base-sroie")
+#philschmid
+pretrained_model = DonutModel.from_pretrained("philschmid/donut-base-sroie")
 #model = VisionEncoderDecoderModel.from_pretrained("sav7669/donut-base-sroie")
 pretrained_model.eval()