Update README.md

README.md

@@ -11,7 +11,7 @@ tags:
 - plants
 ---
 ## Model Overview
-AgroNt is a DNA language model trained on primarily edible plant genomes. More specifically, AgroNT uses the transformer architecture with self-attention and a masked language modeling
+AgroNT is a DNA language model trained on primarily edible plant genomes. More specifically, AgroNT uses the transformer architecture with self-attention and a masked language modeling
 objective to leverage highly available genotype data from 48 different plant speices to learn general representations of nucleotide sequences. AgroNT contains 1 billion parameters and has a context window of 1024 tokens. 
 AgroNt uses a non-overlapping 6-mer tokenizer to convert genomic nucletoide sequences to tokens. As a result the 1024 tokens correspond to approximately 6144 base pairs. 
 
@@ -22,7 +22,7 @@ from transformers import AutoModelForMaskedLM, AutoTokenizer
 import torch
 
 
-model_name = 'agro-nt'
+model_name = 'agro-nucleotide-transformer-1b'
 
 # fetch model and tokenizer from InstaDeep's hf repo
 agro_nt_model = AutoModelForMaskedLM.from_pretrained(f'InstaDeepAI/{model_name}')