Matt C commited on
Commit
79c7e0d
1 Parent(s): 37e34d5
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -26,6 +26,10 @@ toxicProb = prediction.data[0][1]
26
  neutralProb = torch.round(neutralProb, decimals=4)
27
  toxicProb = torch.round(toxicProb, decimals=4)
28
 
 
 
 
 
29
  st.write("Classification Probabilities")
30
- st.write(f"Neutral: {neutralProb:.4}")
31
- st.write(f"Toxic: {toxicProb:.4}")
 
26
  neutralProb = torch.round(neutralProb, decimals=4)
27
  toxicProb = torch.round(toxicProb, decimals=4)
28
 
29
+ # default text input ought to return:
30
+ # Neutral: 0.0052
31
+ # Toxic: 0.9948
32
+
33
  st.write("Classification Probabilities")
34
+ st.write(f"{neutralProb:.4} - NEUTRAL")
35
+ st.write(f"{toxicProb:.4} - TOXIC")