adibvafa
/

CodonTransformer-base

Token Classification

CodonTransformer

Computational Biology

Machine Learning

Synthetic Biology

Model card Files Files and versions

adibvafa commited on Aug 23, 2024

Commit

a66e467

·

verified ·

1 Parent(s): a5a5a1c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -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)
 # Set your input data

 # Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("adibvafa/CodonTransformer")
+model = BigBirdForMaskedLM.from_pretrained("adibvafa/CodonTransformer-base").to(DEVICE)
 # Set your input data