| from spleeter.separator import Separator |
| import gradio as gr |
| import shutil |
|
|
|
|
| def spleeter(aud, instrument): |
| separator = Separator('spleeter:2stems') |
| try: |
| shutil.rmtree("output") |
| except FileNotFoundError: |
| pass |
| separator.separate_to_file(aud.name, "output/", filename_format="audio_example/{instrument}.wav") |
| return f"./output/audio_example/{instrument}.wav", f"./output/audio_example/{instrument}.wav" |
|
|
| inputs = [ |
| gr.Audio(label="Input Audio", type="filepath"), |
| gr.Radio(label="Instrument", choices=["vocals", "accompaniment"], type="value") |
| ] |
| outputs = [ |
| gr.Audio(label="Output Audio", type="filepath"), |
| gr.File(label="Output File") |
| ] |
|
|
| title = "Spleeter" |
| description = "Gradio demo for Spleeter: a fast and efficient music source separation tool with pre-trained models. To use it, simply upload your audio, or click one of the examples to load them. Read more at the links below." |
| article = "<p style='text-align: center'><a href='https://research.deezer.com/projects/spleeter.html'>Spleeter: a Fast and Efficient Music Source Separation Tool with Pre-Trained Models</a> | <a href='https://github.com/deezer/spleeter'>Github Repo</a></p>" |
| examples = [ |
| ["audio_example.mp3", "vocals"] |
| ] |
|
|
| gr.Interface(spleeter, inputs, outputs, title=title, description=description, article=article, examples=examples).launch() |
