File size: 845 Bytes
91f65f4
 
 
 
 
 
7699d3a
91f65f4
 
 
 
 
 
 
4205702
 
 
 
91f65f4
 
 
7699d3a
4205702
91f65f4
 
 
4205702
7699d3a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import os
import gradio as gr


def function(audio_path, output_dir="example/output"):
    filename = os.path.basename(audio_path).split('.')[0]

    cmd_result = os.popen(f'python spleeter separate -p spleeter:2stems -o {output_dir} {audio_path}').read()
    print(cmd_result)
    return f"example/output/{filename}/accompaniment.wav", f"example/output/{filename}/vocals.wav"

inputs = [
    gr.inputs.Audio(source="upload", type="filepath", label=None, optional=False)
]
outputs = [
    gr.outputs.Audio(type="file", label='accompaniment'),
    gr.outputs.Audio(type="file", label='vocals')
]

examples = ["example/audio_example.mp3"]


iface = gr.Interface(fn=function, inputs=inputs, outputs=outputs, examples=examples)
iface.launch(share=True)

if __name__ == "__main__":
    function("example/audio_example.mp3", "example/output")
    pass