amandakonet
commited on
Commit
•
6127b9f
1
Parent(s):
a5675d3
Update README.md
Browse files
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 = ['
|
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 |
```
|