Update app.py

app.py

@@ -16,20 +16,17 @@ import numpy as np
 model_card = ModelCard(
     name="Demucs Stem Separator (All Stems)",
     description=(
-    "Separates a full music track into its core components using the Demucs model. \n"
-    "Input: A stereo music mix (WAV or MP3) containing multiple instruments or vocals.\n"
-    "Output: Four individual audio stems:- Drums, Bass, Vocals, Other-instrumental, each isolated with high fidelity for remixing or analysis. \n\n"
-    "**Model Variants and Recommendations:**\n"
-        "1. **htdemucs**: Best overall realism and clarity (default)\n"
-        "2. **mdx_extra_q**: Highest vocal quality and cleaner separation\n"
-        "3. **mdx_extra**: Balanced trade-off between speed and accuracy\n"
-        "4. **mdx_q**: Lightweight and fast, suitable for quick previews or limited hardware\n\n"
+    "Separates a full music track into four individual audio stems- Drums, Bass, Vocals, Other-instrumental using the Demucs model. \n"
+    "Available Model Variants:\n"
+        "1. htdemucs: Best overall fidelity and natural sound (default) \n"
+        "2. mdx_extra: Balanced trade-off, faster processing with only a small drop in quality. \n"
+        "3. mdx_extra_q: Fastest and lightest, noticeable quality loss, ideal for quick previews. \n"
     ),
     author="Alexandre Défossez, et al.",
     tags=["demucs", "source-separation", "pyharp", "stems", "multi-output"]
 )
 
-DEMUX_MODELS = ["mdx_extra_q", "mdx_extra", "htdemucs", "mdx_q"]
+DEMUX_MODELS = ["htdemucs", "mdx_extra", "mdx_extra_q"]
 STEM_NAMES = ["Drums", "Bass", "Vocals", "Other"]
 
 # Global model cache
@@ -99,6 +96,9 @@ def separate_all_stems(audio_file_path: str, model_name: str):
 
 # Process Function
 def process_fn(audio_file_path, model_name):
+    if model_name not in DEMUX_MODELS:
+        raise ValueError(f"Unsupported model selected: {model_name}")
+
     output_signals = separate_all_stems(audio_file_path, model_name)
 
     is_mp3 = Path(audio_file_path).suffix.lower() == ".mp3"
@@ -124,10 +124,9 @@ with gr.Blocks() as demo:
         value="htdemucs",
         info=(
             "Choose which Demucs variant to use for separation:\n"
-            "1. htdemucs: Best overall quality and realism (default).\n"
-            "2. mdx_extra_q: High-quality extended model trained for cleaner vocals.\n"
-            "3. mdx_extra: Balanced version with good speed and quality.\n"
-            "4. mdx_q: Faster, lighter model for quick previews or low-end devices."
+            "1. htdemucs: High-quality Hybrid Transformer Demucs trained on MusDB + 800 songs.\n"
+            "2. mdx_extra: Trained with extended data. Balanced model offering a clear trade-off between quality and speed.\n"
+            "3. mdx_extra_q: Quantized lightweight version of mdx_extra.\n"
         )
     )