Update README.md

README.md

@@ -28,7 +28,7 @@ The model could be used for protein feature extraction or to be fine-tuned on do
 Here is how to use this model to get the features of a given protein sequence in PyTorch:
 
 ```python
-from transformers import BertModel, BertTokenizer
+from transformers import AutoModel, AutoTokenizer
 
 tokenizer = AutoTokenizer.from_pretrained('qilowoq/AbLang_heavy')
 model = AutoModel.from_pretrained('qilowoq/AbLang_heavy', trust_remote_code=True)