tweak

app.py

@@ -23,9 +23,9 @@ prediction = nn.functional.softmax(result.logits, dim=-1)
 neutralProb = prediction.data[0][0]
 toxicProb = prediction.data[0][1]
 
-neutralProb = torch.round(neutralProb, decimals=3)
-toxicProb = torch.round(toxicProb, decimals=3)
+neutralProb = torch.round(neutralProb, decimals=4)
+toxicProb = torch.round(toxicProb, decimals=4)
 
 st.write("Classification Probabilities")
-st.write(f"Neutral: {neutralProb}")
-st.write(f"Toxic: {toxicProb}")
+st.write(f"Neutral: {neutralProb:.4}")
+st.write(f"Toxic: {toxicProb:.4}")