add Stable Audio option

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 import spaces
+import os
 import json
 import re
 import random
 import numpy as np
 from gradio_client import Client
 
+hf_token = os.environ.get("HF_TOKEN")
 MAX_SEED = np.iinfo(np.int32).max
 
 def check_api(model_name):
@@ -39,6 +41,13 @@ def check_api(model_name):
             return "api ready"
         except : 
             return "api not ready yet"
+    elif model_name == "Stable Audio Open":
+        try:
+            client = Client("fffiloni/Stable-Audio-Open-A10", hf_token=hf_token)
+            return "api ready"
+        except:
+            return "api not ready yet"
+
         
 from moviepy.editor import *
 import cv2
@@ -194,6 +203,18 @@ def get_musicgen(prompt):
     )
     print(result)
     return result[1]
+
+def get_stable_audio_open(prompt):
+    client = Client("fffiloni/Stable-Audio-Open-A10", hf_token=hf_token)
+    result = client.predict(
+		prompt=prompt,
+		seconds_total=10,
+		steps=100,
+		cfg_scale=7,
+		api_name="/predict"
+    )
+    print(result)
+    return result
     
 import re
 import torch
@@ -297,6 +318,9 @@ def infer(video_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_stable_audio_open(musical_prompt)
 
     final_res = blend_vmsc(video_in, music_o)
     return gr.update(value=musical_prompt, interactive=True), gr.update(visible=True), music_o, final_res
@@ -319,6 +343,9 @@ def retry(video_in, 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_stable_audio_open(musical_prompt)
     final_res = blend_vmsc(video_in, music_o)
     return music_o, final_res
 
@@ -365,7 +392,8 @@ with gr.Blocks(css=css) as demo:
                             "AudioLDM-2",
                             "Riffusion",
                             "Mustango",
-                            "MusicGen"
+                            "MusicGen",
+                            "Stable Audio Open"
                         ],
                         value = None,
                         filterable = False
@@ -431,7 +459,6 @@ with gr.Blocks(css=css) as demo:
             result,
             video_o
         ],
-        concurrency_limit = 4
     )
 
 demo.queue(max_size=16).launch(show_api=False)