Update README.md
Browse files
README.md
CHANGED
@@ -67,7 +67,7 @@ def load_hf_model():
|
|
67 |
```python
|
68 |
def get_embedding(dna_sequence):
|
69 |
model, tokenizer, pooler = load_hf_model()
|
70 |
-
tokenized_input = tokenizer(dna_sequence, return_tensors="pt", padding
|
71 |
with torch.no_grad():
|
72 |
output = model(**tokenized_input)
|
73 |
embedding = pooler(output.last_hidden_state, tokenized_input.attention_mask)
|
|
|
67 |
```python
|
68 |
def get_embedding(dna_sequence):
|
69 |
model, tokenizer, pooler = load_hf_model()
|
70 |
+
tokenized_input = tokenizer(dna_sequence, return_tensors="pt", padding=True)
|
71 |
with torch.no_grad():
|
72 |
output = model(**tokenized_input)
|
73 |
embedding = pooler(output.last_hidden_state, tokenized_input.attention_mask)
|