chong.zhang commited on
Commit
032bd6f
·
1 Parent(s): 29e3049
Files changed (1) hide show
  1. app.py +21 -13
app.py CHANGED
@@ -140,31 +140,39 @@ with gr.Blocks() as demo:
140
  # Textbox for custom input
141
  text_input = gr.Textbox(label="Input Text (For Text-to-Music Task)")
142
 
143
- with gr.Row():
144
- default_prompt_buttons = []
145
- for prompt in default_prompts:
146
- button = gr.Button(value=prompt)
147
- button.click(fn=update_text, inputs=None, outputs=textbox)
148
- default_prompt_buttons.append(button)
149
-
150
  audio_input = gr.Audio(label="Input Audio (For Music Continuation Task)",
151
  type="filepath")
152
  audio_input = cut_audio(audio_input, sr, cut_seconds=5)
153
 
154
  with gr.Column():
155
- chorus = gr.Dropdown(["verse", "chorus", "intro", "outro"],
156
  label="Chorus Mode", value="intro")
157
- fast = gr.Checkbox(label="Fast Inference", value=False)
158
- fade_out = gr.Checkbox(label="Apply Fade Out", value=True)
159
- trim = gr.Checkbox(label="Trim Silence", value=False)
160
- output_sample_rate = gr.Dropdown([24000, 48000],
161
- label="Output Sample Rate",
162
  value=48000)
 
 
 
 
 
163
  max_generate_audio_seconds = gr.Slider(10, 300,
164
  label="Max Generated Audio Length (Seconds)",
165
  value=30)
166
 
167
  music_output = gr.Audio(label="Generated Music Result", type="filepath")
 
 
 
 
 
 
 
 
 
 
 
 
 
168
  generate_button = gr.Button("Generate Music")
169
 
170
  generate_button.click(run_inspiremusic,
 
140
  # Textbox for custom input
141
  text_input = gr.Textbox(label="Input Text (For Text-to-Music Task)")
142
 
 
 
 
 
 
 
 
143
  audio_input = gr.Audio(label="Input Audio (For Music Continuation Task)",
144
  type="filepath")
145
  audio_input = cut_audio(audio_input, sr, cut_seconds=5)
146
 
147
  with gr.Column():
148
+ chorus = gr.Dropdown(["intro", "verse", "chorus", "outro"],
149
  label="Chorus Mode", value="intro")
150
+ output_sample_rate = gr.Dropdown([48000, 24000],
151
+ label="Output Audio Sample Rate (Hz)",
 
 
 
152
  value=48000)
153
+ with gr.Row():
154
+ fast = gr.Checkbox(label="Fast Inference", value=False)
155
+ fade_out = gr.Checkbox(label="Apply Fade Out Effect", value=True)
156
+ # trim = gr.Checkbox(label="Trim Silence", value=False)
157
+
158
  max_generate_audio_seconds = gr.Slider(10, 300,
159
  label="Max Generated Audio Length (Seconds)",
160
  value=30)
161
 
162
  music_output = gr.Audio(label="Generated Music Result", type="filepath")
163
+
164
+ with gr.Row():
165
+ default_prompt_buttons = []
166
+ for prompt in default_prompts:
167
+ button = gr.Button(value=prompt)
168
+ button.click(run_inspiremusic,
169
+ inputs = [task, text_input, None, model_name,
170
+ chorus, fast, fade_out, trim,
171
+ output_sample_rate,
172
+ max_generate_audio_seconds],
173
+ outputs = music_output)
174
+ default_prompt_buttons.append(button)
175
+
176
  generate_button = gr.Button("Generate Music")
177
 
178
  generate_button.click(run_inspiremusic,