Update README.md

README.md

@@ -8,7 +8,7 @@ tags:
 
 # Model Card for Mistral-Prot-small (Mistral for protein)
 
-The Mistral-Prot-small Large Language Model (LLM) is a pretrained generative chemical molecule model with 1.9M parameters x 8 experts = 15.2M parameters. 
+The Mistral-Prot-v1-15M Large Language Model (LLM) is a pretrained generative chemical molecule model with 1.9M parameters x 8 experts = 15.2M parameters. 
 It is derived from Mixtral-8x7B-v0.1 model, which was simplified for protein: the number of layers and the hidden size were reduced. 
 The model was pretrained using 1M protein strings from the uniprot 50 database. 
 
@@ -26,8 +26,8 @@ Like Mixtral-8x7B-v0.1, it is a transformer model, with the following architectu
 import torch
 from transformers import AutoTokenizer, AutoModel
 
-tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/Mistral-Prot-small", trust_remote_code=True) 
-model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Prot-small", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/Mistral-Prot-v1-15M", trust_remote_code=True) 
+model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Prot-v1-15M", trust_remote_code=True)
 ```
 
 ## Calculate the embedding of a DNA sequence
@@ -48,7 +48,7 @@ Ensure you are utilizing a stable version of Transformers, 4.34.0 or newer.
 
 ## Notice
 
-Mistral-Prot-small is a pretrained base model for protein.
+Mistral-Prot-v1-15M is a pretrained base model for protein.
 
 ## Contact