Update README.md

README.md

@@ -6,9 +6,9 @@ tags:
 - protein
 ---
 
-# Model Card for Mistral-Prot-v1-139M (Mistral for protein)
+# Model Card for Mistral-Prot-v1-134M (Mistral for protein)
 
-The Mistral-Prot-v1-139M Large Language Model (LLM) is a pretrained generative chemical molecule model with 133.8M parameters. 
+The Mistral-Prot-v1-134M Large Language Model (LLM) is a pretrained generative chemical molecule model with 133.8M 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-v1-139M", trust_remote_code=True) 
-model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Prot-v1-139M", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/Mistral-Prot-v1-134M", trust_remote_code=True) 
+model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Prot-v1-134M", 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-v1-139M is a pretrained base model for protein.
+Mistral-Prot-v1-134M is a pretrained base model for protein.
 
 ## Contact