Update README.md

README.md

@@ -19,8 +19,8 @@ WandB report, https://wandb.ai/mesolitica/llama2-embedding-600m-contrastive/repo
 from transformers import AutoModel, AutoTokenizer
 from sklearn.metrics.pairwise import cosine_similarity
 
-model = AutoModel.from_pretrained('mistral-embedding-191m-8k-contrastive', trust_remote_code = True)
-tokenizer = AutoTokenizer.from_pretrained('mistral-embedding-191m-8k-contrastive')
+tokenizer = AutoTokenizer.from_pretrained("mesolitica/mistral-embedding-191m-8k-contrastive", trust_remote_code=True)
+model = AutoModel.from_pretrained("mesolitica/mistral-embedding-191m-8k-contrastive", trust_remote_code=True)
 
 input_ids = tokenizer(
     [