alisawuffles
/

roberta-large-wanli

Text Classification

Inference Endpoints

Model card Files Files and versions Community

alisawuffles commited on Mar 30, 2022

Commit

05abe3b

•

1 Parent(s): 13599d1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ x = tokenizer("I believe I will get into UW.", "I will get into UW.", hypothesis
 logits = model(**x).logits
 probs = logits.softmax(dim=1).squeeze(0)
 label_id = torch.argmax(probs).item()
-prediction = nli_model.config.id2label[label_id]
 ```
 ### Citation

 logits = model(**x).logits
 probs = logits.softmax(dim=1).squeeze(0)
 label_id = torch.argmax(probs).item()
+prediction = model.config.id2label[label_id]
 ```
 ### Citation