Spaces:
Build error
Build error
removed neutral
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def zs(premise,hypothesis):
|
|
17 |
probs = entail_contradiction_logits.softmax(dim=1)
|
18 |
contra_prob = round(probs[:,0].item() * 100,2)
|
19 |
# neut_prob = round(probs[:,1].item() * 100,2)
|
20 |
-
entail_prob = round(probs[:,
|
21 |
# return contra_prob, neut_prob, entail_prob
|
22 |
return contra_prob, entail_prob
|
23 |
|
|
|
17 |
probs = entail_contradiction_logits.softmax(dim=1)
|
18 |
contra_prob = round(probs[:,0].item() * 100,2)
|
19 |
# neut_prob = round(probs[:,1].item() * 100,2)
|
20 |
+
entail_prob = round(probs[:,1].item() * 100,2)
|
21 |
# return contra_prob, neut_prob, entail_prob
|
22 |
return contra_prob, entail_prob
|
23 |
|