Testing new feature

app.py

@@ -35,8 +35,10 @@ def classify_toxic(text):
         batch = tokenizer.encode(text, return_tensors="pt")
         output = model(batch).logits
     probabilities = torch.nn.functional.softmax(output, dim=-1)
-    probabilities = probabilities.tolist()
-    return "Toxic" if probabilities[0] <= 0.55 else "Safe"
+    preds = probabilities.tolist()
+    print(f"Preds: {preds}")
+    return "Safe"
+    # return "Toxic" if preds[0] <= 0.55 else "Safe"
 
 
 # -----------------------