Update pipe.py

pipe.py

@@ -102,7 +102,7 @@ class SmilesDiffusionPipe(Pipeline):
                 input_ids = input_ids[1:-1]
                 return self.decode(input_ids)
         
-            encoded = self.encode(self.tokenize(masked_smi))
+            encoded = self.encode(self.tokenize_smiles(masked_smi))
             encoded_tensor = torch.tensor(encoded, dtype=torch.long).unsqueeze(0).to(device)
         
             for step in range(steps):