| import os | |
| import tempfile | |
| from audio_separator.separator import Separator | |
| separator = None | |
| def load_model(model = 'MDX23C-8KFFT-InstVoc_HQ.ckpt'): | |
| global separator | |
| separator = Separator() | |
| separator.load_model(model) | |
| def run(audio_path): | |
| global separator | |
| if not audio_path: | |
| return None, None | |
| primary_stem_output_path, secondary_stem_output_path = separator.separate(audio_path) | |
| return primary_stem_output_path, secondary_stem_output_path | |
| def raw(audio): | |
| with tempfile.NamedTemporaryFile(delete=True) as t: | |
| with open(t.name, 'w+b') as f: | |
| f.write(audio) | |
| return run(t.name) | |