nicholasKluge commited on
Commit
ced5451
1 Parent(s): 4d8cb31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,9 +21,9 @@ model.to("cpu")
21
  rewardModel.to(device)
22
  toxicityModel.to(device)
23
 
24
- tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=token)
25
- rewardTokenizer = AutoTokenizer.from_pretrained(rewardmodel_id, use_auth_token=token)
26
- toxiciyTokenizer = AutoTokenizer.from_pretrained(toxicitymodel_id, use_auth_token=token)
27
 
28
 
29
  intro = """
 
21
  rewardModel.to(device)
22
  toxicityModel.to(device)
23
 
24
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
25
+ rewardTokenizer = AutoTokenizer.from_pretrained(rewardmodel_id)
26
+ toxiciyTokenizer = AutoTokenizer.from_pretrained(toxicitymodel_id)
27
 
28
 
29
  intro = """