add MusicGen option

app.py

@@ -158,6 +158,16 @@ def get_mustango(prompt):
     )
     print(result)
     return result
+
+def get_musicgen(prompt):
+    client = Client("https://facebook-musicgen.hf.space/")
+    result = client.predict(
+        prompt,	# str  in 'Describe your music' Textbox component
+        None,	# str (filepath or URL to file) in 'File' Audio component
+        fn_index=0
+    )
+    print(result)
+    return result[1]
     
 import re
 import torch
@@ -239,6 +249,9 @@ def infer(image_in, chosen_model, api_status):
     elif chosen_model == "Mustango" :
         gr.Info("Now calling Mustango for music...")
         music_o = get_mustango(musical_prompt)
+    elif chosen_model == "MusicGen" :
+        gr.Info("Now calling MusicGen for music...")
+        music_o = get_musicgen(musical_prompt)
     
     return gr.update(value=musical_prompt, interactive=True), gr.update(visible=True), music_o
 
@@ -257,11 +270,14 @@ def retry(chosen_model, caption):
     elif chosen_model == "Mustango" :
         gr.Info("Now calling Mustango for music...")
         music_o = get_mustango(musical_prompt)
+    elif chosen_model == "MusicGen" :
+        gr.Info("Now calling MusicGen for music...")
+        music_o = get_musicgen(musical_prompt)
 
     return music_o
 
 demo_title = "Image to Music V2"
-description = "Get music from a picture"
+description = "Get music from a picture, compare text-to-music models"
 
 css = """
 #col-container {
@@ -302,7 +318,8 @@ with gr.Blocks(css=css) as demo:
                             "MAGNet",
                             "AudioLDM-2",
                             "Riffusion",
-                            "Mustango"
+                            "Mustango",
+                            "MusicGen"
                         ],
                         value = None,
                         filterable = False