Update README.md

README.md

@@ -15,11 +15,11 @@ license: "apache-2.0"
 # Demo in a `fill-mask` task
 
 ```
-from transformers import RobertaTokenizerFast, RobertaForMaskedLM, pipeline
+from transformers import AutoTokenizer, AutoModelForMaskedLM, pipeline
 
 model_name = 'sangjeedondrub/tibetan-roberta-base'
-model = AutoModelForMaskedLM.from_pretrained("sangjeedondrub/tibetan-roberta-base")
-tokenizer = AutoTokenizer.from_pretrained("sangjeedondrub/tibetan-roberta-base")
+model = AutoModelForMaskedLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 fill_mask_pipe = pipeline(
     "fill-mask",