Update main.py

main.py

@@ -78,7 +78,7 @@ def run_generate_runner(const_smiles, var_smiles, main_cls, minor_cls, delta_val
         'model_path' :  './raw_pretrain_frag/checkpoint',
         'num_samples' :  50,
         'overwrite' :  True,
-        'save_directory' :  './demo_gen',
+        # 'save_directory' :  './demo_gen',
         'test_file_name' :  'test_cut',
         'vocab_path' :  './'
     }