modified tokenizer
Browse files
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
|
| 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 |
|