JaviSwift commited on
Commit
8a4a02e
·
1 Parent(s): 4aac5af

Fix - deprecated function variable

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,13 +9,13 @@ access_token = os.getenv("HF_ACCESS_TOKEN")
9
  tokenizer = AutoTokenizer.from_pretrained(
10
  MODEL_NAME,
11
  trust_remote_code=True,
12
- use_auth_token=access_token)
13
  model = AutoModelForCausalLM.from_pretrained(
14
  MODEL_NAME,
15
  torch_dtype=torch.bfloat16,
16
  device_map="auto",
17
  trust_remote_code=True,
18
- use_auth_token=access_token
19
  )
20
 
21
  def improve_code(code: str) -> str:
 
9
  tokenizer = AutoTokenizer.from_pretrained(
10
  MODEL_NAME,
11
  trust_remote_code=True,
12
+ token=access_token)
13
  model = AutoModelForCausalLM.from_pretrained(
14
  MODEL_NAME,
15
  torch_dtype=torch.bfloat16,
16
  device_map="auto",
17
  trust_remote_code=True,
18
+ token=access_token
19
  )
20
 
21
  def improve_code(code: str) -> str: