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)
|