Fix toxicchat bug

app.py

@@ -102,7 +102,6 @@ def predict_toxicity(text, model, tokenizer, device, model_name):
             outputs = model.generate(**inputs)
         
         prediction = tokenizer.decode(outputs[0], skip_special_tokens=True).strip().lower()
-        prediction = "Toxic" if prediction == "positive" else "Not Toxic"
     else:
         inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128, padding="max_length").to(device)