Update README.md

README.md

@@ -21,7 +21,7 @@ widget:
 
 ## Model description
 
-[cdsBERT](https://doi.org/10.1101/2023.09.15.558027) is pLM with a codon vocabulary that was seeded with [ProtBERT](https://huggingface.co/Rostlab/prot_bert_bfd) and trained with a novel vocabulary extension pipeline called MELD. cdsBERT offers a highly biologically relevant latent space with excellent EC number prediction surpassing ProtBERT.
+[cdsBERT+](https://doi.org/10.1101/2023.09.15.558027) is pLM with a codon vocabulary that was seeded with [ProtBERT](https://huggingface.co/Rostlab/prot_bert_bfd) and trained with a novel vocabulary extension pipeline called MELD. cdsBERT+ offers a highly biologically relevant latent space with excellent EC number prediction surpassing ProtBERT.
 
 ## How to use
 
@@ -30,7 +30,7 @@ widget:
 import re
 import torch
 import torch.nn.functional as F
-from transformers import BertForMaskedLM, BertTokenizer
+from transformers import BertModel, BertTokenizer
 
 model = BertModel.from_pretrained('lhallee/cdsBERT') # load model
 tokenizer = BertTokenizer.from_pretrained('lhallee/cdsBERT') # load tokenizer