add ACE Step model

app.py

@@ -47,6 +47,12 @@ def check_api(model_name):
             return "api ready"
         except:
             return "api not ready yet"
+    elif model_name == "ACE Step":
+        try :
+            client = Client("ACE-Step/ACE-Step")
+            return "api ready"
+        except : 
+            return "api not ready yet"
     
         
 from moviepy.editor import VideoFileClip
@@ -185,6 +191,39 @@ def get_stable_audio_open(prompt):
     print(result)
     return result
 
+def get_ace(prompt):
+    from gradio_client import Client, handle_file
+
+    client = Client("ACE-Step/ACE-Step")
+    result = client.predict(
+		audio_duration=-1,
+		prompt=prompt,
+		lyrics="[inst]",
+		infer_step=60,
+		guidance_scale=15,
+		scheduler_type="euler",
+		cfg_type="apg",
+		omega_scale=10,
+		manual_seeds=None,
+		guidance_interval=0.5,
+		guidance_interval_decay=0,
+		min_guidance_scale=3,
+		use_erg_tag=True,
+		use_erg_lyric=False,
+		use_erg_diffusion=True,
+		oss_steps=None,
+		guidance_scale_text=0,
+		guidance_scale_lyric=0,
+		audio2audio_enable=False,
+		ref_audio_strength=0.5,
+		ref_audio_input=None,
+		lora_name_or_path="none",
+		api_name="/__call__"
+    )
+    print(result)
+    return result[0]
+
+
 import re
 import torch
 from transformers import pipeline
@@ -272,6 +311,9 @@ def infer(image_in, chosen_model, api_status):
     elif chosen_model == "Stable Audio Open" :
         gr.Info("Now calling Stable Audio Open for music...")
         music_o = get_stable_audio_open(musical_prompt)
+    elif chosen_model == "ACE Step" :
+        gr.Info("Now calling ACE Step for music...")
+        music_o = get_ace(musical_prompt)
     
     return gr.update(value=musical_prompt, interactive=True), gr.update(visible=True), music_o
 
@@ -297,6 +339,9 @@ def retry(chosen_model, caption):
     elif chosen_model == "Stable Audio Open" :
         gr.Info("Now calling Stable Audio Open for music...")
         music_o = get_stable_audio_open(musical_prompt)
+    elif chosen_model == "ACE Step" :
+        gr.Info("Now calling ACE Step for music...")
+        music_o = get_ace(musical_prompt)
 
     return music_o
 
@@ -341,6 +386,7 @@ with gr.Blocks(css=css) as demo:
                         label = "Choose a model",
                         choices = [
                             #"MAGNet",
+                            "ACE Step",
                             "AudioLDM-2",
                             "Riffusion",
                             "Mustango",