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: purple
 colorTo: pink
 sdk: gradio
-sdk_version: 5.28.0
+sdk_version: 6.24.0
 python_version: '3.11'
 app_file: app.py
 pinned: false

app.py

@@ -26,7 +26,7 @@ import torch
 import torchaudio
 from huggingface_hub import hf_hub_download
 from omegaconf import OmegaConf
-from pyharp import ModelCard, build_endpoint
+from pyharp import ModelCard, build_endpoint, Subcategory
 
 # cqt_nsgt_pytorch clips an int-dtype array against np.inf with out=, which
 # numpy's ufunc casting rules reject on any numpy version. Compute without
@@ -258,40 +258,40 @@ model_card = ModelCard(
         "is filled."
     ),
     author="Eloi Moliner, Vesa Välimäki",
-    tags=["audio-inpainting", "diffusion", "restoration"],
+    tags=[Subcategory.INFILLING, 'audio-inpainting', 'diffusion', 'restoration'],
 )
 
 
-with gr.Blocks() as demo:
-    input_components = [
-        gr.Audio(type="filepath", label="Input Audio").harp_required(True),
-        gr.Dropdown(
-            choices=list(CHECKPOINTS.keys()),
-            value="MAESTRO (solo piano)",
-            label="Model",
-            info="Which checkpoint to inpaint with.",
-        ),
-        gr.Slider(minimum=0.05, maximum=3.0, step=0.05, value=1.5, label="Gap Length (s)",
-                  info="Length of the gap to fill (default: 1.5s, per repo config)"),
-        gr.Number(value=-1, label="Gap Position (s)",
-                  info="Where the gap starts, from the beginning of the processed segment. -1 = centered (default, per repo config)"),
-        gr.Slider(minimum=10, maximum=150, step=1, value=35, label="Processing Steps",
-                  info="Diffusion sampling steps (default: T=35, per repo config)"),
-        gr.Slider(minimum=0.0, maximum=1.0, step=0.05, value=0.25, label="Reconstruction Guidance",
-                  info="How strongly the output is pulled toward the surrounding audio (default: ξ=0.25, per repo config)"),
-    ]
-    output_components = [
-        gr.Audio(type="filepath", label="Output Audio").set_info(
-            "The processed segment with the gap filled in."
-        ),
-    ]
-
-    build_endpoint(
-        model_card=model_card,
-        input_components=input_components,
-        output_components=output_components,
-        process_fn=process_fn,
-    )
-
 if __name__ == "__main__":
-    demo.queue().launch(pwa=True)
+    with gr.Blocks() as demo:
+        input_components = [
+            gr.Audio(type="filepath", label="Input Audio").harp_required(True),
+            gr.Dropdown(
+                choices=list(CHECKPOINTS.keys()),
+                value="MAESTRO (solo piano)",
+                label="Model",
+                info="Which checkpoint to inpaint with.",
+            ),
+            gr.Slider(minimum=0.05, maximum=3.0, step=0.05, value=1.5, label="Gap Length (s)",
+                      info="Length of the gap to fill (default: 1.5s, per repo config)"),
+            gr.Number(value=-1, label="Gap Position (s)",
+                      info="Where the gap starts, from the beginning of the processed segment. -1 = centered (default, per repo config)"),
+            gr.Slider(minimum=10, maximum=150, step=1, value=35, label="Processing Steps",
+                      info="Diffusion sampling steps (default: T=35, per repo config)"),
+            gr.Slider(minimum=0.0, maximum=1.0, step=0.05, value=0.25, label="Reconstruction Guidance",
+                      info="How strongly the output is pulled toward the surrounding audio (default: ξ=0.25, per repo config)"),
+        ]
+        output_components = [
+            gr.Audio(type="filepath", label="Output Audio").set_info(
+                "The processed segment with the gap filled in."
+            ),
+        ]
+
+        build_endpoint(
+            model_card=model_card,
+            input_components=input_components,
+            output_components=output_components,
+            process_fn=process_fn,
+        )
+
+    demo.queue().launch(show_error=True, share=True, pwa=True)

requirements.txt

@@ -1,4 +1,4 @@
-git+https://github.com/TEAMuP-dev/pyharp.git@v0.3.0
+git+https://github.com/TEAMuP-dev/pyharp.git@v0.4.0
 cqt_nsgt_pytorch
 einops
 torch
@@ -7,3 +7,4 @@ omegaconf
 soundfile
 numpy
 scipy
+spaces>=0.51.1