Update app.py

app.py

@@ -81,19 +81,19 @@ st.image(image, caption='Your target document')
 
 with st.spinner(f'Processing the document ...'):
         pre_trained_model = "unstructuredio/chipper-fast-fine-tuning"
-        processor = DonutProcessor.from_pretrained(pre_trained_model)
+        processor = DonutProcessor.from_pretrained(pre_trained_model, use_auth_token=os.environ['HF_TOKEN'])
         
         device = "cuda" if torch.cuda.is_available() else "cpu"
 
         if 'model' in st.session_state:
             model = st.session_state['model']
         else:
-            model = VisionEncoderDecoderModel.from_pretrained(pre_trained_model)
+            model = VisionEncoderDecoderModel.from_pretrained(pre_trained_model, use_auth_token=os.environ['HF_TOKEN'])
 
             from huggingface_hub import hf_hub_download
 
             lm_head_file = hf_hub_download(
-                repo_id=pre_trained_model, filename="lm_head.pth"
+                repo_id=pre_trained_model, filename="lm_head.pth", token=os.environ['HF_TOKEN']
             )
 
             rank = 128