Spaces:
Running
Running
added Tango model
Browse files
app.py
CHANGED
|
@@ -117,6 +117,17 @@ def get_audiogen(prompt):
|
|
| 117 |
)
|
| 118 |
return result
|
| 119 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 120 |
def infer(image_in, chosen_model):
|
| 121 |
caption = get_caption(image_in)
|
| 122 |
if chosen_model == "MAGNet" :
|
|
@@ -128,6 +139,9 @@ def infer(image_in, chosen_model):
|
|
| 128 |
elif chosen_model == "AudioGen" :
|
| 129 |
audiogen_result = get_audiogen(caption)
|
| 130 |
return audiogen_result
|
|
|
|
|
|
|
|
|
|
| 131 |
|
| 132 |
css="""
|
| 133 |
#col-container{
|
|
@@ -150,7 +164,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 150 |
with gr.Column():
|
| 151 |
image_in = gr.Image(sources=["upload"], type="filepath", label="Image input", value="oiseau.png")
|
| 152 |
with gr.Row():
|
| 153 |
-
chosen_model = gr.
|
| 154 |
submit_btn = gr.Button("Submit")
|
| 155 |
with gr.Column():
|
| 156 |
audio_o = gr.Audio(label="Audio output")
|
|
|
|
| 117 |
)
|
| 118 |
return result
|
| 119 |
|
| 120 |
+
def get_tango(prompt):
|
| 121 |
+
client = Client("https://declare-lab-tango.hf.space/")
|
| 122 |
+
result = client.predict(
|
| 123 |
+
"Howdy!", # str representing string value in 'Prompt' Textbox component
|
| 124 |
+
100, # int | float representing numeric value between 100 and 200 in 'Steps' Slider component
|
| 125 |
+
4, # int | float representing numeric value between 1 and 10 in 'Guidance Scale' Slider component
|
| 126 |
+
api_name="/predict"
|
| 127 |
+
)
|
| 128 |
+
print(result)
|
| 129 |
+
return result
|
| 130 |
+
|
| 131 |
def infer(image_in, chosen_model):
|
| 132 |
caption = get_caption(image_in)
|
| 133 |
if chosen_model == "MAGNet" :
|
|
|
|
| 139 |
elif chosen_model == "AudioGen" :
|
| 140 |
audiogen_result = get_audiogen(caption)
|
| 141 |
return audiogen_result
|
| 142 |
+
elif chosen_model == "Tango" :
|
| 143 |
+
tango_result = get_tango(caption)
|
| 144 |
+
return tango_result
|
| 145 |
|
| 146 |
css="""
|
| 147 |
#col-container{
|
|
|
|
| 164 |
with gr.Column():
|
| 165 |
image_in = gr.Image(sources=["upload"], type="filepath", label="Image input", value="oiseau.png")
|
| 166 |
with gr.Row():
|
| 167 |
+
chosen_model = gr.Dropdown(label="Choose a model", choices=["MAGNet", "AudioLDM-2", "AudioGen", "Tango"], value="AudioLDM-2")
|
| 168 |
submit_btn = gr.Button("Submit")
|
| 169 |
with gr.Column():
|
| 170 |
audio_o = gr.Audio(label="Audio output")
|