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