Hugging Face's logo

adibvafa
/
CodonTransformer-base

Token Classification
Transformers
Safetensors
big_bird
fill-mask
CodonTransformer
Computational Biology
Machine Learning
Bioinformatics
Synthetic Biology
Model card Files
xet
 Community
adibvafa commited on
Commit
a5a5a1c
·
verified ·
1 Parent(s): 7e1a69d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -30,7 +30,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
30
 
31
  # Load model and tokenizer
32
  tokenizer = AutoTokenizer.from_pretrained("adibvafa/CodonTransformer")
33
- model = BigBirdForMaskedLM.from_pretrained("adibvafa/CodonTransformer").to(DEVICE)
34
 
35
 
36
  # Set your input data
 
30
 
31
  # Load model and tokenizer
32
  tokenizer = AutoTokenizer.from_pretrained("adibvafa/CodonTransformer")
33
+ model = BigBirdForMaskedLM.from_pretrained("adibvafa/CodonTransformer-pretrain").to(DEVICE)
34
 
35
 
36
  # Set your input data