Spaces:
Runtime error
Runtime error
Matt C
commited on
Commit
·
1b0e188
1
Parent(s):
2ce83a2
tweak
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ result = model(batch)
|
|
| 21 |
prediction = nn.functional.softmax(result.logits, dim=-1)
|
| 22 |
|
| 23 |
neutralProb = prediction.data[0]
|
| 24 |
-
toxicProb = prediction.data[
|
| 25 |
|
| 26 |
print("Classification Probabilities")
|
| 27 |
print(f"Neutral: {neutralProb}")
|
|
|
|
| 21 |
prediction = nn.functional.softmax(result.logits, dim=-1)
|
| 22 |
|
| 23 |
neutralProb = prediction.data[0]
|
| 24 |
+
toxicProb = prediction.data[0]
|
| 25 |
|
| 26 |
print("Classification Probabilities")
|
| 27 |
print(f"Neutral: {neutralProb}")
|