File size: 2,947 Bytes
763d88e 8814906 763d88e 8814906 763d88e 8814906 10d4684 763d88e 10d4684 763d88e 10d4684 763d88e 10d4684 763d88e 10d4684 763d88e 8814906 763d88e 8814906 763d88e 8814906 763d88e 8814906 763d88e 8814906 763d88e 8814906 |
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 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
# app.py
import gradio as gr
import torch
import torchaudio
from demucs.pretrained import get_model
from demucs.apply import apply_model
import os
# --- Setup the model ---
print("Setting up the model...")
device = 'cuda' if torch.cuda.is_available() else 'cpu'
print(f"Using device: {device}")
model = get_model(name="htdemucs")
model = model.to(device)
model.eval()
print("Model loaded successfully.")
# --- Separation function ---
def separate_stems(audio_path):
"""
Separates an audio file into drums, bass, other, and vocals.
Returns FILE PATHS (not base64).
"""
if audio_path is None:
return None, None, None, None, "Please upload an audio file."
try:
print(f"Loading audio from: {audio_path}")
wav, sr = torchaudio.load(audio_path)
if wav.shape[0] == 1:
print("Audio is mono, converting to stereo.")
wav = wav.repeat(2, 1)
wav = wav.to(device)
print("Applying the separation model...")
with torch.no_grad():
sources = apply_model(model, wav[None], device=device, progress=True)[0]
print("Separation complete.")
# Save stems temporarily
stem_names = ["drums", "bass", "other", "vocals"]
output_dir = "separated_stems"
os.makedirs(output_dir, exist_ok=True)
output_paths = []
for i, name in enumerate(stem_names):
out_path = os.path.join(output_dir, f"{name}.wav")
torchaudio.save(out_path, sources[i].cpu(), sr)
output_paths.append(out_path)
print(f"✅ Saved {name} to {out_path}")
return output_paths[0], output_paths[1], output_paths[2], output_paths[3], "✅ Separation successful!"
except Exception as e:
print(f"Error: {e}")
return None, None, None, None, f"❌ Error: {str(e)}"
# --- Gradio UI ---
print("Creating Gradio interface...")
with gr.Blocks(theme=gr.themes.Soft()) as demo:
gr.Markdown("# 🎵 Music Stem Separator with HT Demucs")
with gr.Row():
with gr.Column():
audio_input = gr.Audio(type="filepath", label="Upload Your Song")
separate_button = gr.Button("Separate Music", variant="primary")
status_output = gr.Textbox(label="Status", interactive=False)
with gr.Column():
gr.Markdown("### 🎧 Separated Stems")
drums_output = gr.Audio(label="Drums", type="filepath")
bass_output = gr.Audio(label="Bass", type="filepath")
other_output = gr.Audio(label="Other", type="filepath")
vocals_output = gr.Audio(label="Vocals", type="filepath")
separate_button.click(
fn=separate_stems,
inputs=audio_input,
outputs=[drums_output, bass_output, other_output, vocals_output, status_output]
)
gr.Markdown("---\n<p style='text-align: center; font-size: small;'>Powered by HT Demucs</p>")
demo.launch(share=True)
|