Update README.md

README.md

@@ -16,9 +16,9 @@ This is a Japanese BART V2 base model pre-trained on Japanese Wikipedia.
 You can use this model as follows:
 
 ```python
-from transformers import AutoTokenizer, AutoModelForMaskedLM
-tokenizer = AutoTokenizer.from_pretrained('ku-nlp/bart-v2-base-japanese')
-model = AutoModelForMaskedLM.from_pretrained('ku-nlp/bart-v2-base-japanese/')
+from transformers import XLMRobertaTokenizer, MBartForConditionalGeneration
+tokenizer = XLMRobertaTokenizer.from_pretrained('ku-nlp/bart-v2-base-japanese')
+model = MBartForConditionalGeneration.from_pretrained('ku-nlp/bart-v2-base-japanese/')
 sentence = '京都 大学 で 自然 言語 処理 を 専攻 する 。'  # input should be segmented into words by Juman++ in advance
 encoding = tokenizer(sentence, return_tensors='pt')
 ...