Spaces:
Runtime error
Runtime error
Matt C
commited on
Commit
•
3fea615
1
Parent(s):
d72e17f
tweak
Browse files
app.py
CHANGED
@@ -23,8 +23,8 @@ prediction = nn.functional.softmax(result.logits, dim=-1)
|
|
23 |
neutralProb = prediction.data[0][0]
|
24 |
toxicProb = prediction.data[0][1]
|
25 |
|
26 |
-
neutralProb = torch.round(neutralProb, 3)
|
27 |
-
toxicProb = torch.round(toxicProb, 3)
|
28 |
|
29 |
st.write("Classification Probabilities")
|
30 |
st.write(f"Neutral: {neutralProb}")
|
|
|
23 |
neutralProb = prediction.data[0][0]
|
24 |
toxicProb = prediction.data[0][1]
|
25 |
|
26 |
+
neutralProb = torch.round(neutralProb, decimals=3)
|
27 |
+
toxicProb = torch.round(toxicProb, decimals=3)
|
28 |
|
29 |
st.write("Classification Probabilities")
|
30 |
st.write(f"Neutral: {neutralProb}")
|