Added missing imports in README.md

README.md

@@ -46,6 +46,9 @@ Inputs require SELFIES tokenization **and** a precomputed distance matrix
 (`relative_position`). Use the helper bundled in the repo:
 
 ```python
+import torch
+from transformers import AutoTokenizer
+
 tokenizer = AutoTokenizer.from_pretrained("IlPakoZ/m5-encoder", trust_remote_code=True)
 
 smiles = "CCO"