tweak

app.py

@@ -26,6 +26,10 @@ toxicProb = prediction.data[0][1]
 neutralProb = torch.round(neutralProb, decimals=4)
 toxicProb = torch.round(toxicProb, decimals=4)
 
+# default text input ought to return:
+# Neutral: 0.0052
+# Toxic: 0.9948
+
 st.write("Classification Probabilities")
-st.write(f"Neutral: {neutralProb:.4}")
-st.write(f"Toxic: {toxicProb:.4}")
+st.write(f"{neutralProb:.4}  -  NEUTRAL")
+st.write(f"{toxicProb:.4}  -  TOXIC")