Update README.md

README.md

@@ -24,15 +24,13 @@ This repository provides a base-sized Japanese RoBERTa model. The model was trai
 
 # How to load the model
 
-*NOTE:* Use `T5Tokenizer` to initiate the tokenizer.
-
 ~~~~
-from transformers import T5Tokenizer, RobertaForMaskedLM
+from transformers import AutoTokenizer, AutoModelForMaskedLM
 
-tokenizer = T5Tokenizer.from_pretrained("rinna/japanese-roberta-base")
+tokenizer = AutoTokenizer.from_pretrained("rinna/japanese-roberta-base", use_fast=False)
 tokenizer.do_lower_case = True  # due to some bug of tokenizer config loading
 
-model = RobertaForMaskedLM.from_pretrained("rinna/japanese-roberta-base")
+model = AutoModelForMaskedLM.from_pretrained("rinna/japanese-roberta-base")
 ~~~~
 
 # How to use the model for masked token prediction