Update src/mdx.py
Browse files- 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}.
|
| 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}.
|
| 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:
|