Spaces:
Runtime error
Runtime error
generate audio minor refactor
Browse files
app.py
CHANGED
|
@@ -172,17 +172,20 @@ def generate_audio(prompt: str, duration: int) -> gr.Component:
|
|
| 172 |
Returns:
|
| 173 |
gr.Component: The generated audio as a Gradio Audio component.
|
| 174 |
"""
|
|
|
|
|
|
|
|
|
|
| 175 |
client = Tool.from_space(
|
| 176 |
space_id="luke9705/MusicGen_custom",
|
| 177 |
token=os.environ.get('HF_TOKEN'),
|
| 178 |
name="Sound_Generator",
|
| 179 |
description="Generate music or sound effects from a text prompt using MusicGen."
|
| 180 |
)
|
| 181 |
-
if duration > 30:
|
| 182 |
-
sound = client(prompt, 30)
|
| 183 |
-
else:
|
| 184 |
-
sound = client(prompt, duration)
|
| 185 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 186 |
return gr.Audio(value=sound)
|
| 187 |
|
| 188 |
@tool
|
|
|
|
| 172 |
Returns:
|
| 173 |
gr.Component: The generated audio as a Gradio Audio component.
|
| 174 |
"""
|
| 175 |
+
|
| 176 |
+
DURATION_LIMIT = 30
|
| 177 |
+
|
| 178 |
client = Tool.from_space(
|
| 179 |
space_id="luke9705/MusicGen_custom",
|
| 180 |
token=os.environ.get('HF_TOKEN'),
|
| 181 |
name="Sound_Generator",
|
| 182 |
description="Generate music or sound effects from a text prompt using MusicGen."
|
| 183 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 184 |
|
| 185 |
+
duration = duration if duration < DURATION_LIMIT else DURATION_LIMIT
|
| 186 |
+
|
| 187 |
+
sound = client(prompt, duration)
|
| 188 |
+
|
| 189 |
return gr.Audio(value=sound)
|
| 190 |
|
| 191 |
@tool
|