Update app.py

app.py

@@ -6,13 +6,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequen
 model_id = "nicholasKluge/Aira-Instruct-PT-124M"
 rewardmodel_id = "nicholasKluge/RewardModelPT"
 toxicitymodel_id = "nicholasKluge/ToxicityModelPT"
-token = "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"
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-model = AutoModelForCausalLM.from_pretrained(model_id, use_auth_token=token)
-rewardModel = AutoModelForSequenceClassification.from_pretrained(rewardmodel_id, use_auth_token=token)
-toxicityModel = AutoModelForSequenceClassification.from_pretrained(toxicitymodel_id, use_auth_token=token)
+model = AutoModelForCausalLM.from_pretrained(model_id)
+rewardModel = AutoModelForSequenceClassification.from_pretrained(rewardmodel_id)
+toxicityModel = AutoModelForSequenceClassification.from_pretrained(toxicitymodel_id)
 
 model.eval()
 rewardModel.eval()