Update inference/style_transfer.py

inference/style_transfer.py

@@ -78,9 +78,9 @@ class Mixing_Style_Transfer_Inference:
                 separate_file_names.append(args.reference_file_name_2interpolate)
             for cur_idx, cur_inf_dir in enumerate(sorted(glob(f"{args.target_dir}*/"))):
                 for cur_file_name in separate_file_names:
-                    cur_sep_file_path = os.path.join(cur_inf_dir, cur_file_name+'.mp3')
+                    cur_sep_file_path = os.path.join(cur_inf_dir, cur_file_name+'.wav')
                     cur_sep_output_dir = os.path.join(cur_inf_dir, args.stem_level_directory_name)
-                    if os.path.exists(os.path.join(cur_sep_output_dir, self.args.separation_model, cur_file_name, 'drums.mp3')):
+                    if os.path.exists(os.path.join(cur_sep_output_dir, self.args.separation_model, cur_file_name, 'drums.wav')):
                         print(f'\talready separated current file : {cur_sep_file_path}')
                     else:
                         cur_cmd_line = f"demucs {cur_sep_file_path} -n {self.args.separation_model} -d {self.device} -o {cur_sep_output_dir}"