Update README.md

README.md

@@ -8,9 +8,9 @@ tags:
 - genomics
 ---
 
-# Model Card for mixtral-dna-yeast-v0.2 (mistral for DNA)
+# Model Card for Mistral-DNA-v1-138M-yeast (mistral for DNA)
 
-The mixtral-dna-yeast-v0.2 Large Language Model (LLM) is a pretrained generative DNA text model with 17.31M parameters x 8 experts = 138.5M parameters. 
+The Mistral-DNA-v1-138M-yeast Large Language Model (LLM) is a pretrained generative DNA text model with 17.31M parameters x 8 experts = 138.5M parameters. 
 It is derived from Mistral-7B-v0.1 model, which was simplified for DNA: the number of layers and the hidden size were reduced. 
 The model was pretrained using around 1000 yeast genomes with 10kb DNA sequences. 
 
@@ -31,8 +31,8 @@ Like Mistral-7B-v0.1, it is a transformer model, with the following architecture
 import torch
 from transformers import AutoTokenizer, AutoModel
 
-tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/mixtral-dna-yeast-v0.2", trust_remote_code=True) # Same as DNABERT2
-model = AutoModel.from_pretrained("RaphaelMourad/mixtral-dna-yeast-v0.2", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/Mistral-DNA-v1-138M-yeast", trust_remote_code=True) # Same as DNABERT2
+model = AutoModel.from_pretrained("RaphaelMourad/Mistral-DNA-v1-138M-yeast", trust_remote_code=True)
 ```
 
 ## Calculate the embedding of a DNA sequence
@@ -53,7 +53,7 @@ Ensure you are utilizing a stable version of Transformers, 4.34.0 or newer.
 
 ## Notice
 
-Mistral-DNA is a pretrained base model for DNA.
+Mistral-DNA-v1-138M-yeast is a pretrained base model for DNA.
 
 ## Contact