Spaces:
Running
Running
File size: 1,605 Bytes
420f095 b1dbfe9 00f8cdc 420f095 b1dbfe9 2891437 b1dbfe9 420f095 |
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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import os
import gradio as gr
from scipy.io.wavfile import write, read
import subprocess
def inference(audio):
os.makedirs("out", exist_ok=True)
write('mix.wav', audio[0], audio[1])
command = "python3 -m demucs -n mdx_extra_q -d cpu mix.wav -o out"
process = subprocess.run(command,
shell=True,
stdin=subprocess.DEVNULL,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE
)
# Check if files exist before returning
files = ["./out/mdx_extra_q/mix/vocals.wav",
"./out/mdx_extra_q/mix/bass.wav",
"./out/mdx_extra_q/mix/drums.wav",
"./out/mdx_extra_q/mix/other.wav"]
for file in files:
if not os.path.isfile(file):
print(f"File not found: {file}")
else:
print(f"File exists: {file}")
return files
article = "Inspired by <p><a href='https://github.com/facebookresearch/demucs' target='_blank'>Demucs</a></p>\n<p>Copyright © 2023 JP Madsen</p"
demo = gr.Interface(
inference,
gr.inputs.Audio(type = "numpy", label = "Input"),
[gr.outputs.Audio(type = "filepath", label = "Vocals"),
gr.outputs.Audio(type = "filepath", label = "Bass"),
gr.outputs.Audio(type = "filepath", label = "Drums"),
gr.outputs.Audio(type = "filepath", label = "Other")
],
article = article,
theme = 'NoCrypt/[email protected]',
allow_flagging = "never",
css="style.css", # Hinzugefügt
)
#demo.queue(concurrency_count = 10)
demo.launch()
|