Update README.md

README.md

@@ -65,8 +65,8 @@ import torch
 import torch.nn.functional as F
 
 # Load model (trust_remote_code=True required for custom model)
-tokenizer = AutoTokenizer.from_pretrained("nikhil061307/contrastive-clinical-entity-model")
-model = AutoModel.from_pretrained("nikhil061307/contrastive-clinical-entity-model", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("nikhil061307/contrastive-learning-bert-added-token-v5")
+model = AutoModel.from_pretrained("nikhil061307/contrastive-learning-bert-added-token-v5", trust_remote_code=True)
 
 def get_clinical_embeddings(texts, max_length=256):
     """Get embeddings for clinical texts with [ENTITY] support."""