|
|
import demucs.separate |
|
|
import os |
|
|
def demucs_split( |
|
|
url_or_path: str, |
|
|
demucs_model: str = "mdx_extra" |
|
|
): |
|
|
|
|
|
input_file = "input.mp3" |
|
|
|
|
|
def clean_temp(): |
|
|
clear_folder("splited_audio_output") |
|
|
|
|
|
input_file = url_or_path |
|
|
|
|
|
print("Separating audio...") |
|
|
os.environ['XDG_CACHE_HOME'] = "./models_demucs/" |
|
|
demucs_opts = ["--mp3", "--two-stems", "vocals", "-n", demucs_model, input_file, "-o", "splited_audio_output", "--filename", "{stem}.mp3"] |
|
|
demucs.separate.main(demucs_opts) |
|
|
|
|
|
return f"splited_audio_output/{demucs_model}/vocals.mp3", f"splited_audio_output/{demucs_model}/no_vocals.mp3" |
