yigitturali commited on
Commit
6e15aca
·
verified ·
1 Parent(s): 2efa49a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = True)
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)