Spaces:
Running
Running
re-enable audiogen
Browse files
app.py
CHANGED
|
@@ -108,12 +108,13 @@ def get_audioldm(prompt):
|
|
| 108 |
|
| 109 |
def get_audiogen(prompt):
|
| 110 |
try:
|
| 111 |
-
client = Client("
|
| 112 |
result = client.predict(
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
| 116 |
)
|
|
|
|
| 117 |
return result
|
| 118 |
except:
|
| 119 |
raise gr.Error("AudioGen space API is not ready, please try again in few minutes ")
|
|
@@ -249,7 +250,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 249 |
chosen_model = gr.Dropdown(label="Choose a model", choices=[
|
| 250 |
#"MAGNet",
|
| 251 |
"AudioLDM-2",
|
| 252 |
-
|
| 253 |
"Tango",
|
| 254 |
#"Tango 2",
|
| 255 |
"Stable Audio Open",
|
|
|
|
| 108 |
|
| 109 |
def get_audiogen(prompt):
|
| 110 |
try:
|
| 111 |
+
client = Client("fffiloni/audiogen", hf_token=hf_token)
|
| 112 |
result = client.predict(
|
| 113 |
+
prompt=prompt,
|
| 114 |
+
duration=10,
|
| 115 |
+
api_name="/infer"
|
| 116 |
)
|
| 117 |
+
print(result)
|
| 118 |
return result
|
| 119 |
except:
|
| 120 |
raise gr.Error("AudioGen space API is not ready, please try again in few minutes ")
|
|
|
|
| 250 |
chosen_model = gr.Dropdown(label="Choose a model", choices=[
|
| 251 |
#"MAGNet",
|
| 252 |
"AudioLDM-2",
|
| 253 |
+
"AudioGen",
|
| 254 |
"Tango",
|
| 255 |
#"Tango 2",
|
| 256 |
"Stable Audio Open",
|