Update README.md
Browse files
README.md
CHANGED
@@ -34,8 +34,8 @@ model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Codon-v1-16M", trust_re
|
|
34 |
## Calculate the embedding of a coding sequence
|
35 |
|
36 |
```
|
37 |
-
|
38 |
-
inputs = tokenizer(
|
39 |
hidden_states = model(inputs)[0] # [1, sequence_length, 256]
|
40 |
|
41 |
# embedding with max pooling
|
|
|
34 |
## Calculate the embedding of a coding sequence
|
35 |
|
36 |
```
|
37 |
+
codon_dna = "TGA TGA TTG GCG CGG CTA GGA TCG GCT"
|
38 |
+
inputs = tokenizer(codon_dna, return_tensors = 'pt')["input_ids"]
|
39 |
hidden_states = model(inputs)[0] # [1, sequence_length, 256]
|
40 |
|
41 |
# embedding with max pooling
|