second attempt

app.py

@@ -107,7 +107,7 @@ def predict_toxicity(text, model, tokenizer, device, model_name):
         print(f"Raw model output: {prediction}")
         
         # Adjust the condition to check for both "positive" and potential variations
-        prediction = "Toxic" if prediction in ["positive", "pos", "toxic", "yes"] else "Not Toxic"
+        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)