hohdavid commited on
Commit
dc957a5
·
verified ·
1 Parent(s): f57f9be

modified tokenizer

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ class ProteinEmbedder:
72
  seq_spaced = " ".join(list(seq))
73
 
74
  with torch.no_grad():
75
- ids = self.tokenizer.batch_encode_plus([seq_spaced], add_special_tokens=True, padding=True, return_tensors="pt")
76
  input_ids = ids['input_ids'].to(self.device)
77
  attention_mask = ids['attention_mask'].to(self.device)
78
 
 
72
  seq_spaced = " ".join(list(seq))
73
 
74
  with torch.no_grad():
75
+ ids = self.tokenizer([seq_spaced], add_special_tokens=True, padding=True, return_tensors="pt")
76
  input_ids = ids['input_ids'].to(self.device)
77
  attention_mask = ids['attention_mask'].to(self.device)
78