amandakonet commited on
Commit
6127b9f
1 Parent(s): a5675d3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -25,7 +25,7 @@ features = tokenizer(['Beginning in 2005, however, polar ice modestly receded fo
25
  model.eval()
26
  with torch.no_grad():
27
  scores = model(**features).logits
28
- label_mapping = ['contradiction', 'entailment', 'neutral']
29
  labels = [label_mapping[score_max] for score_max in scores.argmax(dim=1)]
30
  print(labels)
31
  ```
 
25
  model.eval()
26
  with torch.no_grad():
27
  scores = model(**features).logits
28
+ label_mapping = ['entailment', 'contradiction', 'neutral']
29
  labels = [label_mapping[score_max] for score_max in scores.argmax(dim=1)]
30
  print(labels)
31
  ```