update README

README.md

@@ -18,27 +18,27 @@ This repository is based on the following open source projects: [google-research
 #### How to use
 
 ```python
-from transformers import pipeline, AutoTokenizer, BertForMaskedLM
+from transformers import pipeline, AutoTokenizer, AutoModelForMaskedLM
 
 tokenizer = AutoTokenizer.from_pretrained('tugstugi/bert-base-mongolian-cased')
-model = BertForMaskedLM.from_pretrained('tugstugi/bert-base-mongolian-cased')
+model = AutoModelForMaskedLM.from_pretrained('tugstugi/bert-base-mongolian-cased')
 
 ## declare task ##
 pipe = pipeline(task="fill-mask", model=model, tokenizer=tokenizer)
 
 ## example ##
-input_  = 'Миний [MASK] хоол идэх нь тун чухал.'
+input_  = '[MASK] хот Монгол улсын нийслэл.'
 
 output_ = pipe(input_)
 for i in range(len(output_)):
     print(output_[i])
 
-## Output ##
-# {'sequence': 'Улаанбаатар хотын Сонгино хайрхан дүүрэг.', 'score': 0.9701077342033386, 'token': 281, 'token_str': 'Улаанбаатар'}
-# {'sequence': 'УБ хотын Сонгино хайрхан дүүрэг.', 'score': 0.02008666843175888, 'token': 7389, 'token_str': 'УБ'}
-# {'sequence': 'Нийслэл хотын Сонгино хайрхан дүүрэг.', 'score': 0.006682577542960644, 'token': 4059, 'token_str': 'Нийслэл'}
-# {'sequence': 'Улаанбаатар хотын Сонгино хайрхан дүүрэг.', 'score': 0.0008267111843451858, 'token': 2328, 'token_str': 'Улаанбаатар'}
-# {'sequence': 'Улаанбаатарын хотын Сонгино хайрхан дүүрэг.', 'score': 0.0003509577363729477, 'token': 5593, 'token_str': 'Улаанбаатарын'}
+## output ##
+# {'sequence': 'Улаанбаатар хот Монгол улсын нийслэл.', 'score': 0.826970100402832, 'token': 281, 'token_str': 'Улаанбаатар'}
+# {'sequence': 'Нийслэл хот Монгол улсын нийслэл.', 'score': 0.06551621109247208, 'token': 4059, 'token_str': 'Нийслэл'}
+# {'sequence': 'Эрдэнэт хот Монгол улсын нийслэл.', 'score': 0.0264141745865345, 'token': 2229, 'token_str': 'Эрдэнэт'}
+# {'sequence': 'Дархан хот Монгол улсын нийслэл.', 'score': 0.017083868384361267, 'token': 1646, 'token_str': 'Дархан'}
+# {'sequence': 'УБ хот Монгол улсын нийслэл.', 'score': 0.010854342952370644, 'token': 7389, 'token_str': 'УБ'}
 ```