Add Stable Audio Open option

app.py

@@ -39,6 +39,13 @@ def check_api(model_name):
             return "api ready"
         except : 
             return "api not ready yet"
+    elif model_name == "Stable Audio Open":
+        try:
+            client = Client("artificialguybr/Stable-Audio-Open-Zero")
+            return "api ready"
+        except:
+            return "api not ready yet"
+    
         
 from moviepy.editor import VideoFileClip
 from moviepy.audio.AudioClip import AudioClip
@@ -179,7 +186,19 @@ def get_musicgen(prompt):
     )
     print(result)
     return result[1]
-    
+
+def get_stable_audio_open(prompt):
+    client = Client("artificialguybr/Stable-Audio-Open-Zero")
+    result = client.predict(
+		prompt=prompt,
+		seconds_total=30,
+		steps=100,
+		cfg_scale=7,
+		api_name="/predict"
+    )
+    print(result)
+    return result
+
 import re
 import torch
 from transformers import pipeline
@@ -263,6 +282,9 @@ def infer(image_in, chosen_model, api_status):
     elif chosen_model == "MusicGen" :
         gr.Info("Now calling MusicGen for music...")
         music_o = get_musicgen(musical_prompt)
+    elif chosen_model == "Stable Audio Open" :
+        gr.Info("Now calling Stable Audio Open for music...")
+        music_o = get_sable_audio_open(musocal_prompt)
     
     return gr.update(value=musical_prompt, interactive=True), gr.update(visible=True), music_o
 
@@ -284,6 +306,9 @@ def retry(chosen_model, caption):
     elif chosen_model == "MusicGen" :
         gr.Info("Now calling MusicGen for music...")
         music_o = get_musicgen(musical_prompt)
+    elif chosen_model == "Stable Audio Open" :
+        gr.Info("Now calling Stable Audio Open for music...")
+        music_o = get_sable_audio_open(musocal_prompt)
 
     return music_o
 
@@ -334,7 +359,8 @@ with gr.Blocks(css=css) as demo:
                             "AudioLDM-2",
                             "Riffusion",
                             "Mustango",
-                            "MusicGen"
+                            "MusicGen",
+                            "Stable Audio Open"
                         ],
                         value = None,
                         filterable = False