Update app.py

app.py

@@ -38,7 +38,7 @@ rewardmodel_id = "nicholasKluge/RewardModel"
 toxicitymodel_id = "nicholasKluge/ToxicityModel"
 
 # specify the device (cuda if available)
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # "cuda" if torch.cuda.is_available() else "cpu"
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # load the models (chatbot, reward model, toxicity model)
 model = AutoModelForCausalLM.from_pretrained(model_id)