Update pipe.py

pipe.py

@@ -92,15 +92,15 @@ class SmilesDiffusionPipe(Pipeline):
         mask_token = voc.vocab["[MASK]"]
 
         total_tokens = sequences.size(1)
-            original_num_masked = (sequences == mask_token).sum().item()
-            print(f"Original number of masked tokens: {original_num_masked}")
-    
-            if total_tokens == 0:
-                raise ValueError("Encoded SMILES has no tokens.")
-            if original_num_masked == 0:
-                print("No masked tokens found. Returning original SMILES.")
-                input_ids = input_ids[1:-1]
-                return self.decode(input_ids)
+        original_num_masked = (sequences == mask_token).sum().item()
+        print(f"Original number of masked tokens: {original_num_masked}")
+
+        if total_tokens == 0:
+            raise ValueError("Encoded SMILES has no tokens.")
+        if original_num_masked == 0:
+            print("No masked tokens found. Returning original SMILES.")
+            input_ids = input_ids[1:-1]
+            return self.decode(input_ids)
     
         encoded = self.encode(self.tokenize(masked_smi))
         encoded_tensor = torch.tensor(encoded, dtype=torch.long).unsqueeze(0).to(device)