rachith commited on
Commit
f89b0a7
·
1 Parent(s): 2ed6184

removed neutral

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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[:,2].item() * 100,2)
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