Update README.md
Browse files
README.md
CHANGED
@@ -44,7 +44,7 @@ shift= 1
|
|
44 |
tok_params = {'kmer' : kmer,
|
45 |
'shift' : shift}
|
46 |
tokenizer = ProkBERTTokenizer(tokenization_params=tok_params)
|
47 |
-
model =
|
48 |
sequence = 'CACCGCATGGAGATCGGCACCTACTTCGACAAGCTGGAGGCGCTGCTGAAGGAGTGGTACGAGGCGCGCGGGGGTGAGGCATGACGGACTGGCAAGAGGAGCAGCGTCAGCGC'
|
49 |
inputs = tokenizer(sequence, return_tensors="pt")
|
50 |
# Ensure that inputs have a batch dimension
|
|
|
44 |
tok_params = {'kmer' : kmer,
|
45 |
'shift' : shift}
|
46 |
tokenizer = ProkBERTTokenizer(tokenization_params=tok_params)
|
47 |
+
model = MegatronBertForSequenceClassification.from_pretrained(finetuned_model)
|
48 |
sequence = 'CACCGCATGGAGATCGGCACCTACTTCGACAAGCTGGAGGCGCTGCTGAAGGAGTGGTACGAGGCGCGCGGGGGTGAGGCATGACGGACTGGCAAGAGGAGCAGCGTCAGCGC'
|
49 |
inputs = tokenizer(sequence, return_tensors="pt")
|
50 |
# Ensure that inputs have a batch dimension
|