ajimeno commited on
Commit
6bbcef4
1 Parent(s): c3949f9

Update app.py

Browse files

Updated to the token parameter in from_pretrained

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -81,14 +81,14 @@ st.image(image, caption='Your target document')
81
 
82
  with st.spinner(f'Processing the document ...'):
83
  pre_trained_model = "unstructuredio/chipper-fast-fine-tuning"
84
- processor = DonutProcessor.from_pretrained(pre_trained_model, use_auth_token=os.environ['HF_TOKEN'])
85
 
86
  device = "cuda" if torch.cuda.is_available() else "cpu"
87
 
88
  if 'model' in st.session_state:
89
  model = st.session_state['model']
90
  else:
91
- model = VisionEncoderDecoderModel.from_pretrained(pre_trained_model, use_auth_token=os.environ['HF_TOKEN'])
92
 
93
  from huggingface_hub import hf_hub_download
94
 
 
81
 
82
  with st.spinner(f'Processing the document ...'):
83
  pre_trained_model = "unstructuredio/chipper-fast-fine-tuning"
84
+ processor = DonutProcessor.from_pretrained(pre_trained_model, token=os.environ['HF_TOKEN'])
85
 
86
  device = "cuda" if torch.cuda.is_available() else "cpu"
87
 
88
  if 'model' in st.session_state:
89
  model = st.session_state['model']
90
  else:
91
+ model = VisionEncoderDecoderModel.from_pretrained(pre_trained_model, token=os.environ['HF_TOKEN'])
92
 
93
  from huggingface_hub import hf_hub_download
94