Fixed core issue, I think

app.py

@@ -37,8 +37,7 @@ def classify_toxic(text):
     probabilities = torch.nn.functional.softmax(output, dim=-1)
     preds = probabilities.tolist()
     print(f"Preds: {preds}")
-    return "Safe"
-    # return "Toxic" if preds[0] <= 0.55 else "Safe"
+    return "Toxic" if preds[0][0] <= 0.55 else "Safe"
 
 
 # -----------------------