Bring the Space up to the pyharp example standard (taxonomy tags, __main__ guard, show_error, v0.4.0)

README.md

@@ -4,7 +4,7 @@ emoji: 🥁
 colorFrom: red
 colorTo: purple
 sdk: gradio
-sdk_version: 5.28.0
+sdk_version: 6.24.0
 app_file: app.py
 pinned: false
 license: cc-by-nc-4.0

app.py

@@ -14,6 +14,7 @@ from typing import Union, Tuple, Optional
 from torch import Tensor
 
 from pyharp import build_endpoint, ModelCard
+from pyharp.tags import Subcategory
 
 
 # ─────────────────────────────────────────────
@@ -286,7 +287,7 @@ model_card = ModelCard(
     name="LarsNet Drum Stem Separator",
     description="Separates a drum mix into individual drum stems: Kick, Snare, Toms, Hi-Hat, and Cymbals.",
     author="A. I. Mezza, et al.",
-    tags=["drums", "demucs", "source-separation", "pyharp", "stems", "multi-output"],
+    tags=[Subcategory.STEM_SEPARATION, 'drums', 'demucs', 'source-separation', 'pyharp', 'stems', 'multi-output'],
 )
 
 MODEL = LarsNet(wiener_filter=False, device="cpu", config="config.yaml")
@@ -305,19 +306,20 @@ def process_fn(audio_path: str):
     return tuple(output_paths)
 
 
-with gr.Blocks() as demo:
-    input_audio = gr.Audio(type="filepath", label="Drum Mix (Input)").harp_required(True)
-    output_kick    = gr.Audio(type="filepath", label="Kick")
-    output_snare   = gr.Audio(type="filepath", label="Snare")
-    output_toms    = gr.Audio(type="filepath", label="Toms")
-    output_hihat   = gr.Audio(type="filepath", label="Hi-Hat")
-    output_cymbals = gr.Audio(type="filepath", label="Cymbals")
-
-    app = build_endpoint(
-        model_card=model_card,
-        input_components=[input_audio],
-        output_components=[output_kick, output_snare, output_toms, output_hihat, output_cymbals],
-        process_fn=process_fn,
-    )
+if __name__ == "__main__":
+    with gr.Blocks() as demo:
+        input_audio = gr.Audio(type="filepath", label="Drum Mix (Input)").harp_required(True)
+        output_kick    = gr.Audio(type="filepath", label="Kick")
+        output_snare   = gr.Audio(type="filepath", label="Snare")
+        output_toms    = gr.Audio(type="filepath", label="Toms")
+        output_hihat   = gr.Audio(type="filepath", label="Hi-Hat")
+        output_cymbals = gr.Audio(type="filepath", label="Cymbals")
+
+        app = build_endpoint(
+            model_card=model_card,
+            input_components=[input_audio],
+            output_components=[output_kick, output_snare, output_toms, output_hihat, output_cymbals],
+            process_fn=process_fn,
+        )
 
-demo.queue().launch(show_error=True)
+    demo.queue().launch(pwa=True, share=True, show_error=True)

requirements.txt

@@ -1,8 +1,9 @@
 # pip install -r requirements.txt
-git+https://github.com/TEAMuP-dev/pyharp.git@v0.3.0
+git+https://github.com/TEAMuP-dev/pyharp.git@v0.4.0
 torch==2.7
 torchaudio==2.7
 pyyaml
 tqdm
 soundfile
 torchcodec==0.7
+spaces>=0.51.1