Update app.py

app.py

@@ -109,6 +109,17 @@ def get_audioldm(prompt):
     audio_result = extract_audio(result)
     return audio_result
 
+def get_riffusion(prompt):
+    client = Client("https://fffiloni-spectrogram-to-music.hf.space/--replicas/1qwjx/")
+    result = client.predict(
+		prompt,	# str  in 'Musical prompt' Textbox component
+		"",	# str  in 'Negative prompt' Textbox component
+		"",	# filepath  in 'parameter_4' Audio component
+		10,	# float (numeric value between 5 and 10) in 'Duration in seconds' Slider component
+		api_name="/predict"
+    )
+    print(result)
+    return result[1]
     
 import re
 import torch
@@ -157,6 +168,9 @@ def infer(image_in, chosen_model):
     elif chosen_model == "AudioLDM-2" :
         gr.Info("Now calling AudioLDM-2 for music...")
         music_o = get_magnet(musical_prompt)
+    elif chosen_model == "Riffusion" :
+        gr.Info("Now calling Riffusion for music...")
+        music_o = get_riffusion(musical_prompt)
     
     return musical_prompt, music_o
 
@@ -185,11 +199,12 @@ with gr.Blocks(css=css) as demo:
                     type = "filepath",
                     elem_id = "image-in"
                 )
-                chosen_model = gr.Radio(
+                chosen_model = gr.Dropdown(
                     label = "Choose a model",
                     choices = [
                         "MAGNet",
-                        "AudioLDM-2"
+                        "AudioLDM-2",
+                        "Riffusion"
                     ],
                     value = "MAGNet"
                 )