rodrisouza commited on
Commit
e8e0475
·
verified ·
1 Parent(s): 47d879c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,7 +51,7 @@ def load_model(model_name):
51
  tokenizer = AutoTokenizer.from_pretrained(
52
  models[model_name],
53
  padding_side='left',
54
- use_auth_token=hugging_face_token,
55
  trust_remote_code=True
56
  )
57
 
@@ -62,7 +62,7 @@ def load_model(model_name):
62
 
63
  model = AutoModelForCausalLM.from_pretrained(
64
  models[model_name],
65
- use_auth_token=hugging_face_token,
66
  trust_remote_code=True
67
  )
68
 
 
51
  tokenizer = AutoTokenizer.from_pretrained(
52
  models[model_name],
53
  padding_side='left',
54
+ token=hugging_face_token,
55
  trust_remote_code=True
56
  )
57
 
 
62
 
63
  model = AutoModelForCausalLM.from_pretrained(
64
  models[model_name],
65
+ token=hugging_face_token,
66
  trust_remote_code=True
67
  )
68