Add Satable Audio Open option

app.py

@@ -150,6 +150,22 @@ def get_tango2(prompt):
     )
     print(result)
     return result
+
+def get_stable_audio_open(prompt):
+    try:
+        client = Client("artificialguybr/Stable-Audio-Open-Zero")
+    except:
+        raise gr.Error("Stable Audio Open space API is not ready, please try again in few minutes ")
+    
+    result = client.predict(
+		prompt=prompt,
+		seconds_total=30,
+		steps=100,
+		cfg_scale=7,
+		api_name="/predict"
+    )
+    print(result)
+    return result
     
 def infer(image_in, chosen_model):
     caption = get_caption(image_in)
@@ -168,6 +184,9 @@ def infer(image_in, chosen_model):
     elif chosen_model == "Tango 2" :
         tango2_result = get_tango2(caption)
         return tango2_result
+    elif chosen_model == "Stable Audio Open" :
+        stable_audio_open_result = get_stable_audio_open(caption)
+        return stable_audio_open_result
 
 css="""
 #col-container{
@@ -190,7 +209,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             image_in = gr.Image(sources=["upload"], type="filepath", label="Image input", value="oiseau.png")
             with gr.Row():
-                chosen_model = gr.Dropdown(label="Choose a model", choices=["MAGNet", "AudioLDM-2", "AudioGen", "Tango", "Tango 2"], value="AudioLDM-2")
+                chosen_model = gr.Dropdown(label="Choose a model", choices=["MAGNet", "AudioLDM-2", "AudioGen", "Tango", "Tango 2", "Stable Audio Open"], value="AudioLDM-2")
                 submit_btn = gr.Button("Submit")
         with gr.Column():
             audio_o = gr.Audio(label="Audio output")