helloWorld199 commited on
Commit
885d63c
1 Parent(s): 6576158

Update src/mdx.py

Browse files
Files changed (1) hide show
  1. src/mdx.py +2 -2
src/mdx.py CHANGED
@@ -274,14 +274,14 @@ def run_mdx(model_params, output_dir, model_path, filename, exclude_main=False,
274
 
275
  main_filepath = None
276
  if not exclude_main:
277
- main_filepath = os.path.join(output_dir, f"{os.path.basename(os.path.splitext(filename)[0])}_{stem_name}{_stemname1}.wav")
278
  sf.write(main_filepath, wave_processed.T, sr)
279
 
280
  invert_filepath = None
281
  if not exclude_inversion:
282
  diff_stem_name = stem_naming.get(stem_name) if invert_suffix is None else invert_suffix
283
  stem_name = f"{stem_name}_diff" if diff_stem_name is None else diff_stem_name
284
- invert_filepath = os.path.join(output_dir, f"{os.path.basename(os.path.splitext(filename)[0])}_{stem_name}{_stemname2}.wav")
285
  sf.write(invert_filepath, (-wave_processed.T * model.compensation) + wave.T, sr)
286
 
287
  if not keep_orig:
 
274
 
275
  main_filepath = None
276
  if not exclude_main:
277
+ main_filepath = os.path.join(output_dir, f"{os.path.basename(os.path.splitext(filename)[0])}_{stem_name}{_stemname1}.mp3")
278
  sf.write(main_filepath, wave_processed.T, sr)
279
 
280
  invert_filepath = None
281
  if not exclude_inversion:
282
  diff_stem_name = stem_naming.get(stem_name) if invert_suffix is None else invert_suffix
283
  stem_name = f"{stem_name}_diff" if diff_stem_name is None else diff_stem_name
284
+ invert_filepath = os.path.join(output_dir, f"{os.path.basename(os.path.splitext(filename)[0])}_{stem_name}{_stemname2}.mp3")
285
  sf.write(invert_filepath, (-wave_processed.T * model.compensation) + wave.T, sr)
286
 
287
  if not keep_orig: