Update README.md

README.md

@@ -10,11 +10,15 @@ InformBERT is pretrained model trained using variable masking strategy, where in
 
 More detail: https://arxiv.org/abs/2210.11771
 
-## How to use
+## How to load
 ```Python
-from transformers import AutoTokenizer, AutoModel
-tokenizer = AutoTokenizer.from_pretrained("nsadeq/InformBERT")
+from transformers import BertTokenizer, AutoModel
+tokenizer = BertTokenizer.from_pretrained("nsadeq/InformBERT")
 model = AutoModel.from_pretrained("nsadeq/InformBERT")
+
+from transformers import pipeline
+unmasker = pipeline('fill-mask', model='nsadeq/InformBERT',tokenizer=tokenizer)
+unmasker("SpeedWeek is an American television program on [MASK].")
 ```
 
 ## Citation