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
|