Update app.py

app.py

@@ -1,60 +1,39 @@
 import gradio as gr
 from spleeter.separator import Separator
 from spleeter.audio.adapter import AudioAdapter
-import os
 import tempfile
+import os
 
-# Path to the pretrained models inside your HF Space
 MODEL_PATH = "pretrained_models/2stems"
 
-# Load Spleeter once (much faster for multiple users)
-separator = Separator(f"{MODEL_PATH}")
-
+separator = Separator(MODEL_PATH)
 audio_loader = AudioAdapter.default()
 
-
 def separate_audio(audio_file):
     if audio_file is None:
-        return None, None, "Please upload an audio file."
+        return None, None, "Upload an audio file."
 
-    # Load audio
     waveform, sample_rate = audio_loader.load(audio_file)
-
-    # Separate stems
     prediction = separator.separate(waveform)
 
-    # Temporary output files
-    tmp_dir = tempfile.mkdtemp()
-    vocal_path = os.path.join(tmp_dir, "vocals.wav")
-    instrumental_path = os.path.join(tmp_dir, "instrumental.wav")
-
-    # Save outputs
-    audio_loader.save(vocal_path, prediction["vocals"], sample_rate)
-    audio_loader.save(instrumental_path, prediction["accompaniment"], sample_rate)
-
-    return vocal_path, instrumental_path, "Done!"
-
-
-# Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎵 Vocal & Instrumental Separator (Spleeter 2-Stems)")
-    gr.Markdown("Upload a song and get **vocals** and **instrumental** extracted!")
-
-    with gr.Row():
-        input_audio = gr.Audio(type="filepath", label="Upload Audio")
-
-    with gr.Row():
-        vocal_output = gr.Audio(label="Vocals")
-        instru_output = gr.Audio(label="Instrumental")
+    tmpdir = tempfile.mkdtemp()
+    vocals_path = os.path.join(tmpdir, "vocals.wav")
+    instr_path = os.path.join(tmpdir, "instrumental.wav")
 
-    status = gr.Textbox(label="Status")
+    audio_loader.save(vocals_path, prediction["vocals"], sample_rate)
+    audio_loader.save(instr_path, prediction["accompaniment"], sample_rate)
 
-    run_btn = gr.Button("Separate")
+    return vocals_path, instr_path, "Done!"
 
-    run_btn.click(
-        separate_audio,
-        inputs=input_audio,
-        outputs=[vocal_output, instru_output, status]
-    )
+iface = gr.Interface(
+    fn=separate_audio,
+    inputs=gr.Audio(type="filepath", label="Upload Audio"),
+    outputs=[
+        gr.Audio(label="Vocals"),
+        gr.Audio(label="Instrumental"),
+        gr.Textbox(label="Status")
+    ],
+    title="🎵 Spleeter Vocal/Instrumental Separator"
+)
 
-demo.launch()
+iface.launch()