Update app.py

app.py

@@ -10,9 +10,11 @@ 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
+
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
+huggingface_hub.notebook_login(token=HF_TOKEN)
 
-#huggingface_hub.notebook_login(token=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)
 
 def demo_process(input_img):
     global pretrained_model, task_prompt, task_name
@@ -33,8 +35,8 @@ image.save("sroie_test2.jpg")
 
 #philschmid
 pretrained_model = DonutModel.from_pretrained("philschmid/donut-base-sroie")
-model = VisionEncoderDecoderModel.from_pretrained("philschmid/donut-base-sroie")
-model.eval()
+#model = VisionEncoderDecoderModel.from_pretrained("philschmid/donut-base-sroie")
+pretrained_model.eval()
 
 demo = gr.Interface(
     fn=demo_process,