Update README.md

README.md

@@ -21,3 +21,27 @@ between bulgarian and Bulgarian. The training data is Bulgarian text from [OSCAR
 It was finetuned on private NLI Bulgarian data.
 
 Then, it was compressed via [progressive module replacing](https://arxiv.org/abs/2002.02925).
+
+### How to use
+
+Here is how to use this model in PyTorch:
+
+```python
+>>> import torch
+>>> from transformers import AutoModelForSequenceClassification, AutoTokenizer
+>>> 
+>>> model_id = 'rmihaylov/bert-base-nli-theseus-bg'
+>>> model = AutoModelForSequenceClassification.from_pretrained(model_id)
+>>> tokenizer = AutoTokenizer.from_pretrained(model_id)
+>>>
+>>> inputs = tokenizer.encode_plus(
+>>>     'Няколко момчета играят футбол.', 
+>>>     'Няколко момичета играят футбол.', 
+>>>     return_tensors='pt')
+>>>
+>>> outputs = model(**inputs)
+>>> contradiction, entailment, neutral = torch.softmax(outputs[0][0], dim=0).detach()
+>>> contradiction, neutral, entailment
+
+(tensor(0.9998), tensor(0.0001), tensor(5.9929e-05))
+```