Update app.py

app.py

@@ -120,6 +120,17 @@ def get_riffusion(prompt):
     )
     print(result)
     return result[1]
+
+def get_mustango(prompt):
+    client = Client("https://declare-lab-mustango.hf.space/")
+    result = client.predict(
+		"Hello!!",	# str  in 'Prompt' Textbox component
+		100,	# float (numeric value between 100 and 200) in 'Steps' Slider component
+		1,	# float (numeric value between 1 and 10) in 'Guidance Scale' Slider component
+							api_name="/predict"
+    )
+    print(result)
+    return result
     
 import re
 import torch
@@ -171,6 +182,9 @@ def infer(image_in, chosen_model):
     elif chosen_model == "Riffusion" :
         gr.Info("Now calling Riffusion for music...")
         music_o = get_riffusion(musical_prompt)
+    elif chosen_model == "Mustango" :
+        gr.Info("Now calling Mustango for music...")
+        music_o = get_mustango(musical_prompt)
     
     return musical_prompt, music_o
 
@@ -204,7 +218,8 @@ with gr.Blocks(css=css) as demo:
                     choices = [
                         "MAGNet",
                         "AudioLDM-2",
-                        "Riffusion"
+                        "Riffusion",
+                        "Mustango"
                     ],
                     value = "MAGNet"
                 )