AutoModelForMaskedLM preferred

README.md

@@ -20,7 +20,7 @@ This is a RoBERTa model pre-trained on Classical Chinese texts, derived from [Gu
 ## How to Use
 
 ```py
-from transformers import AutoTokenizer,AutoModel
+from transformers import AutoTokenizer,AutoModelForMaskedLM
 tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/roberta-classical-chinese-base-char")
-model=AutoModel.from_pretrained("KoichiYasuoka/roberta-classical-chinese-base-char")
+model=AutoModelForMaskedLM.from_pretrained("KoichiYasuoka/roberta-classical-chinese-base-char")
 ```