add Tango 2 model

app.py

@@ -136,6 +136,21 @@ def get_tango(prompt):
     print(result)
     return result
 
+def get_tango2(prompt):
+    try:
+        client = Client("declare-lab/tango2")
+    except:
+        raise gr.Error("Tango2 space API is not ready, please try again in few minutes ")
+    
+    result = client.predict(
+    		prompt=prompt,
+    		steps=100,
+    		guidance=4,
+    		api_name="/predict"
+    )
+    print(result)
+    return result
+    
 def infer(image_in, chosen_model):
     caption = get_caption(image_in)
     if chosen_model == "MAGNet" :
@@ -150,6 +165,9 @@ def infer(image_in, chosen_model):
     elif chosen_model == "Tango" :
         tango_result = get_tango(caption)
         return tango_result
+    elif chosen_model == "Tango 2" :
+        tango2_result = get_tango2(caption)
+        return tango2_result
 
 css="""
 #col-container{
@@ -172,7 +190,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"], value="AudioLDM-2")
+                chosen_model = gr.Dropdown(label="Choose a model", choices=["MAGNet", "AudioLDM-2", "AudioGen", "Tango", "Tango 2"], value="AudioLDM-2")
                 submit_btn = gr.Button("Submit")
         with gr.Column():
             audio_o = gr.Audio(label="Audio output")