Update README.md

README.md

@@ -57,4 +57,24 @@ tokenizer = AutoTokenizer.from_pretrained("imranraad/autotrain-magpie-epie-combi
 inputs = tokenizer("I love AutoTrain", return_tensors="pt")
 
 outputs = model(**inputs)
+```
+
+How to get the logits:
+
+```
+from transformers import AutoTokenizer, AutoModelForTokenClassification
+from transformers import pipeline
+
+model = AutoModelForTokenClassification.from_pretrained("imranraad/idiom-xlm-roberta")
+
+tokenizer = AutoTokenizer.from_pretrained("imranraad/idiom-xlm-roberta")
+
+pipeline_idioms = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
+
+text = "Why are you so bent out of shape? - Why are you so upset?"
+
+idioms = pipeline_idioms(text)
+for idiom in idioms:
+    if idiom['entity_group'] == '1':
+        print(idiom['word'])
 ```