Update README.md
Browse files
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
|