Update inference/style_transfer.py

inference/style_transfer.py

@@ -171,7 +171,10 @@ class Mixing_Style_Transfer_Inference:
                     sf.write(os.path.join(cur_out_dir, f"{cur_inst_name}_{output_name_tag}.wav"), fin_data_out_inst.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
             # remix
             fin_data_out_mix = sum(inst_outputs)
-            sf.write(os.path.join(cur_out_dir, f"mixture_{output_name_tag}.wav"), fin_data_out_mix.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
+            fin_output_path = os.path.join(cur_out_dir, f"mixture_{output_name_tag}.wav")
+            sf.write(fin_output_path, fin_data_out_mix.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
+
+            return fin_output_path
 
 
     # Inference whole song
@@ -326,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='./samples/style_transfer/')
+    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')