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
|