agentlans commited on
Commit
5bcab17
·
verified ·
1 Parent(s): 5f910da

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -2
README.md CHANGED
@@ -90,7 +90,6 @@ scores = model.predict(
90
  label_mapping = ["entailment", "neutral", "contradiction"]
91
  labels = [label_mapping[score_max] for score_max in scores.argmax(axis=1)]
92
  print(labels)
93
- # Output: ['entailment', 'contradiction']
94
  ```
95
 
96
  ## Using Transformers Library
@@ -119,7 +118,6 @@ with torch.no_grad():
119
  label_mapping = ["entailment", "neutral", "contradiction"]
120
  labels = [label_mapping[score_max] for score_max in scores.argmax(dim=1)]
121
  print(labels)
122
- # Output: ['entailment', 'contradiction']
123
  ```
124
 
125
  ## Limitations and Ethical Considerations
 
90
  label_mapping = ["entailment", "neutral", "contradiction"]
91
  labels = [label_mapping[score_max] for score_max in scores.argmax(axis=1)]
92
  print(labels)
 
93
  ```
94
 
95
  ## Using Transformers Library
 
118
  label_mapping = ["entailment", "neutral", "contradiction"]
119
  labels = [label_mapping[score_max] for score_max in scores.argmax(dim=1)]
120
  print(labels)
 
121
  ```
122
 
123
  ## Limitations and Ethical Considerations