harp

app.py

@@ -375,7 +375,7 @@ def ui_harp(launch_kwargs):
                 with gr.Row():
                     melody = gr.Audio(label="Melody Audio", type="filepath")
                     text = gr.Text(label="Input Text", interactive=True)
-                                    with gr.Row():
+                                    # with gr.Row():
                     # model = gr.Radio(["melody", "medium", "small", "large"],
                     #                  label="Model", value="small", interactive=True)
 
@@ -392,35 +392,31 @@ def ui_harp(launch_kwargs):
 
         def predict_harp(_melody, _text, _duration, _temperature, _cfg_coef):
             from audiotools import AudioSignal
-            import torcb
-            sig = AudioSignal(_melody)
+            import torch
+            # sig = AudioSignal(_melody)
 
-            samples = sig.samples[0].numpy()
+            # samples = (sig.sample_rate, sig.samples[0].numpy())
             output = predict_full(
                 model="small", 
+                melody=None,
                 text=_text, duration=_duration, 
                 topk=250, topp=0, 
                 temperature=_temperature, 
                 cfg_coef=_cfg_coef
             )
-            sig = AudioSignal(torch.from_numpy(output).unsqueeze(0))
-            sig.write('output.wav')
-            return sig.path_to_file
+            return output
             
-        
-        submit.click(predict_harp,
-                     inputs=inputs,
-                     outputs=[output])
-
 
         from pyharp import ModelCard, build_endpoint
         card = ModelCard(
             name="MusicGen (Meta)",
-            description="The model will generate a short music extract based on the description you provided. The model can generate up to 30 seconds of audio in one pass.",
+            description="The model will generate a short excerpt based on the text description you provided. The model can generate up to 30 seconds of audio in one pass.",
             author="Jade Copet et al.",
             tags=["example", "music generation"]
         )
 
+        build_endpoint(inputs, output, predict_harp, card=card)
+
 
         interface.queue().launch(**launch_kwargs)
 

requirements.txt

@@ -19,4 +19,4 @@ demucs
 librosa
 gradio_client==0.2.6
 descript-audiotools
--e git+https://github.com/audacitorch/pyharp.git#egg=pyharp
+-e git+https://github.com/audacitorch/pyharp.git#egg=pyharp