Update README.md

README.md

@@ -22,7 +22,7 @@ To use it on a sentence :
 ````python
 import torch
 
-sentence = "The goal of life is [MASK]."
+sentence = "Let's have a [MASK]."
 
 encoded_inputs = tokenizer([sentence], padding='longest')
 input_ids = torch.tensor(encoded_inputs['input_ids'])
@@ -35,3 +35,15 @@ predicted_token = tokenizer.decode(masked_token)
 
 print(predicted_token)
 ````
+
+Or we can also predict the n most relevant predictions :
+
+````python
+top_n = 5
+
+vocab_size = model.config.vocab_size
+logits = output['logits'][0][mask_index].tolist()
+top_tokens = sorted(list(range(vocab_size)), key=lambda  i:logits[i], reverse=True)[:top_n]
+
+tokenizer.decode(top_tokens)
+````