update

app.py

@@ -114,7 +114,16 @@ def cut_audio(audio_file, cut_seconds=5):
 	torchaudio.save(output_path, cutted_audio, sr)
 	return output_path
 
-def run_inspiremusic(task, text, audio, model_name, chorus, fast, fade_out,
+def run_text2music(task, text, model_name, chorus, fast, fade_out,
+					 output_sample_rate, max_generate_audio_seconds):
+	args = get_args(
+			task=task, text=text, None,
+			model_name=model_name, chorus=chorus, fast=fast,
+			fade_out=fade_out, output_sample_rate=output_sample_rate,
+			max_generate_audio_seconds=max_generate_audio_seconds)
+	return music_generation(args)
+
+def run_continuation(task, text, audio, model_name, chorus, fast, fade_out,
 					 output_sample_rate, max_generate_audio_seconds):
 	args = get_args(
 			task=task, text=text, audio=cut_audio(audio, cut_seconds=5),
@@ -161,21 +170,29 @@ with gr.Blocks() as demo:
 		default_prompt_buttons = []
 		for prompt in default_prompts:
 			button = gr.Button(value=prompt)
-			button.click(run_inspiremusic,
-			inputs = [task, text_input, None, model_name,
-					  chorus, fast, fade_out,
-					  output_sample_rate,
-					  max_generate_audio_seconds],
-			outputs = music_output)
+			button.click(run_text2music,
+						inputs = [task, text_input, None, model_name,
+								  chorus, fast, fade_out,
+								  output_sample_rate,
+								  max_generate_audio_seconds],
+						outputs = music_output)
 			default_prompt_buttons.append(button)
-
-	generate_button = gr.Button("Generate Music")
-
-	generate_button.click(run_inspiremusic,
+	with gr.Row():
+		generate_button = gr.Button("Music Continuation")
+		generate_button.click(run_continuation,
 						  inputs=[task, text_input, audio_input, model_name,
 								  chorus, fast, fade_out,
 								  output_sample_rate,
 								  max_generate_audio_seconds],
 						  outputs=music_output)
+		button = gr.Button("Text to Music")
+
+		button.click(run_text2music,
+						  inputs=[task, text_input, model_name,
+								  chorus, fast, fade_out,
+								  output_sample_rate,
+								  max_generate_audio_seconds],
+						  outputs=music_output)
+
 
 demo.launch()