Update README.md

README.md

@@ -4,7 +4,7 @@ widget:
     - text: "Ḣ Q V Q [MASK] E"
 ---
 
-## AntiBERTa2 🧬
+## AntiBERTa2-CSSP 🧬
 
 AntiBERTa2 is an antibody-specific language model based on the [RoFormer model](https://arxiv.org/abs/2104.09864) - it is pre-trained using masked language modelling.
 We also provide a multimodal version of AntiBERTa2, AntiBERTa2-CSSP, that has been trained using a contrastive objective, similar to the [CLIP method](https://arxiv.org/abs/2103.00020).
@@ -26,8 +26,8 @@ non-commercial use. For any users seeking commercial use of our model and genera
         RoFormerTokenizer, 
         RoFormerForSequenceClassification
     )
->>> tokenizer = RoFormerTokenizer.from_pretrained("alchemab/antiberta2")
->>> model = RoFormerModel.from_pretrained("alchemab/antiberta2")
+>>> tokenizer = RoFormerTokenizer.from_pretrained("alchemab/antiberta2-cssp")
+>>> model = RoFormerModel.from_pretrained("alchemab/antiberta2-cssp")
 >>> model(**tokenizer("Ḣ Q V Q ... T V S S", return_tensors='pt')).last_hidden_state... # etc
 
 >>> new_model = RoFormerForSequenceClassification.from_pretrained(