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+'.wav')
+                    cur_sep_file_path = os.path.join(cur_inf_dir, cur_file_name+'.mp3')
                     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.wav')):
+                    if os.path.exists(os.path.join(cur_sep_output_dir, self.args.separation_model, cur_file_name, 'drums.mp3')):
                         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}"
@@ -329,7 +329,7 @@ def set_up():
 
     directory_args = parser.add_argument_group('Directory args')
     # directory paths
-    directory_args.add_argument('--target_dir', type=str, default='./')
+    directory_args.add_argument('--target_dir', type=str, default='./yt_dir/')
     directory_args.add_argument('--output_dir', type=str, default=None, help='if no output_dir is specified (None), the results will be saved inside the target_dir')
     directory_args.add_argument('--input_file_name', type=str, default='input')
     directory_args.add_argument('--reference_file_name', type=str, default='reference')