Update README.md

README.md

@@ -1,10 +1,23 @@
 ---
 widget:
- - text: "It is quite likely that we will build a space station that allows people to live and work in space. </s></s> We will build a space station that allows people to live and work in space."
+ - text: "I believe I will get into UW. </s></s> I will get into UW."
 ---
 
 This is an off-the-shelf roberta-large model finetuned on WANLI, the Worker-AI Collaborative NLI dataset ([Liu et al., 2022](https://arxiv.org/abs/2201.05955)). It outperforms the `roberta-large-mnli` model on seven out-of-domain test sets, including by 11% on HANS and 9% on Adversarial NLI.
 
+### How to use
+```
+model = RobertaForSequenceClassification.from_pretrained('alisawuffles/roberta-large-wanli')
+tokenizer = RobertaTokenizer.from_pretrained('alisawuffles/roberta-large-wanli')
+
+x = tokenizer("I believe I will get into UW.", "I will get into UW.", hypothesis, return_tensors='pt', max_length=128, truncation=True)
+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
 ```
 @misc{liu-etal-2022-wanli,
     title = "WANLI: Worker and AI Collaboration for Natural Language Inference Dataset Creation",