update

app.py

@@ -114,19 +114,19 @@ def cut_audio(audio_file, cut_seconds=5):
 	torchaudio.save(output_path, cutted_audio, sr)
 	return output_path
 
-def run_text2music(task, text, model_name, chorus, fast, fade_out,
+def run_text2music(text, model_name, chorus, fast, fade_out,
 					 output_sample_rate, max_generate_audio_seconds):
 	args = get_args(
-			task=task, text=text, audio=None,
+			task='continuation', text=text, audio=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,
+def run_continuation(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),
+			task='text-to-music', text=text, audio=cut_audio(audio, cut_seconds=5),
 			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)
@@ -165,7 +165,7 @@ with gr.Blocks() as demo:
 	with gr.Row():
 		button = gr.Button("Text to Music")
 		button.click(run_text2music,
-						  inputs=[task, text_input, model_name,
+						  inputs=[text_input, model_name,
 								  chorus, fast, fade_out,
 								  output_sample_rate,
 								  max_generate_audio_seconds],
@@ -173,7 +173,7 @@ with gr.Blocks() as demo:
 
 		generate_button = gr.Button("Music Continuation")
 		generate_button.click(run_continuation,
-						  inputs=[task, text_input, audio_input, model_name,
+						  inputs=[text_input, audio_input, model_name,
 								  chorus, fast, fade_out,
 								  output_sample_rate,
 								  max_generate_audio_seconds],
@@ -184,7 +184,7 @@ with gr.Blocks() as demo:
 		for prompt in default_prompts:
 			button = gr.Button(value=prompt)
 			button.click(run_text2music,
-						inputs = [task, text_input, model_name,
+						inputs = [text_input, model_name,
 								  chorus, fast, fade_out,
 								  output_sample_rate,
 								  max_generate_audio_seconds],