Update README.md

README.md

@@ -30,7 +30,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("adibvafa/CodonTransformer")
-model = BigBirdForMaskedLM.from_pretrained("adibvafa/CodonTransformer-pretrain").to(DEVICE)
+model = BigBirdForMaskedLM.from_pretrained("adibvafa/CodonTransformer-base").to(DEVICE)
 
 
 # Set your input data