Update README.md

README.md

@@ -19,8 +19,8 @@ KBioXLM model can be fintuned on downstream tasks. The downstream tasks here ref
 You can follow the prompts below to load our model parameters:
 
 ```python
-from transformers import AutoTokenizer, AutoModelForMaskedLM
-model=RobertaModel.from_pretrained(pretrained_model_name_or_path='ngwlh/KBioXLM', config=self.config)
+from transformers import RobertaModel
+model=RobertaModel.from_pretrained('ngwlh/KBioXLM')
 checkpoint = torch.load('ngwlh/KBioXLM'+'/pytorch_model.bin', map_location='cpu')
 new_state_dict=OrderedDict()
 for k,v in checkpoint.items():