DMMP commited on
Commit
1dff562
·
verified ·
1 Parent(s): 7069760

generate audio minor refactor

Browse files
Files changed (1) hide show
  1. app.py +7 -4
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