dwb2023 commited on
Commit
b50be2b
1 Parent(s): cd39699

Update app.py

Browse files

update login info

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import spaces
5
  from transformers import AutoModel
6
 
7
  hf_token = os.getenv("HF_TOKEN")
 
8
 
9
  @spaces.GPU
10
  def get_model_summary(model_name):
 
5
  from transformers import AutoModel
6
 
7
  hf_token = os.getenv("HF_TOKEN")
8
+ login(token=hf_token, add_to_git_credential=True)
9
 
10
  @spaces.GPU
11
  def get_model_summary(model_name):