etrop commited on
Commit
239d34a
1 Parent(s): ca6ec69

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -31,9 +31,9 @@ agro_nt_tokenizer = AutoTokenizer.from_pretrained(f'InstaDeepAI/{model_name}')
31
  print(f"Loaded the {model_name} model with {agro_nt_model.num_parameters()} parameters and corresponding tokenizer.")
32
 
33
  # example sequence and tokenization
34
- sequences = ['ATATACGGCCGNC']
35
 
36
- batch_tokens = agro_nt_tokenizer(sequences)['input_ids']
37
  print(f"Tokenzied sequence: {agro_nt_tokenizer.batch_decode(batch_tokens)}")
38
 
39
  torch_batch_tokens = torch.tensor(batch_tokens)
 
31
  print(f"Loaded the {model_name} model with {agro_nt_model.num_parameters()} parameters and corresponding tokenizer.")
32
 
33
  # example sequence and tokenization
34
+ sequences = ['ATATACGGCCGNC','GGGTATCGCTTCCGAC']
35
 
36
+ batch_tokens = agro_nt_tokenizer(sequences,padding="longest")['input_ids']
37
  print(f"Tokenzied sequence: {agro_nt_tokenizer.batch_decode(batch_tokens)}")
38
 
39
  torch_batch_tokens = torch.tensor(batch_tokens)