update

app.py

@@ -18,8 +18,13 @@ def generate_filename():
 
 def get_args(
 		task, text="", audio=None, model_name="InspireMusic-Base",
-		chorus="intro", fast=False, fade_out=True,
+		chorus="intro",
 		output_sample_rate=48000, max_generate_audio_seconds=30.0, time_start = 0.0, time_end=30.0, trim=False):
+	
+	if output_sample_rate == 24000:
+		fast = True
+	else:
+		fast = False
 	# This function constructs the arguments required for InspireMusic
 	args = {
 		"task"                      : task,
@@ -28,7 +33,7 @@ def get_args(
 		"model_name"                : model_name,
 		"chorus"                    : chorus,
 		"fast"                      : fast,
-		"fade_out"                  : fade_out,
+		"fade_out"                  : True,
 		"trim"                      : trim,
 		"output_sample_rate"        : output_sample_rate,
 		"min_generate_audio_seconds": 10.0,
@@ -99,25 +104,24 @@ def cut_audio(audio_file, cut_seconds=5):
 	torchaudio.save(output_path, cutted_audio, sr)
 	return output_path
 
-def run_text2music(text, model_name, chorus, fast, fade_out,
+def run_text2music(text, model_name, chorus,
 					 output_sample_rate, max_generate_audio_seconds):
 	args = get_args(
 			task='continuation', text=text, audio=None,
-			model_name=model_name, chorus=chorus, fast=fast,
-			fade_out=fade_out, output_sample_rate=output_sample_rate,
+			model_name=model_name, chorus=chorus,
+			output_sample_rate=output_sample_rate,
 			max_generate_audio_seconds=max_generate_audio_seconds)
 	return music_generation(args)
 
-def run_continuation(text, audio, model_name, chorus, fast, fade_out,
+def run_continuation(text, audio, model_name, chorus,
 					 output_sample_rate, max_generate_audio_seconds):
 	args = get_args(
 			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,
+			model_name=model_name, chorus=chorus,
+			output_sample_rate=output_sample_rate,
 			max_generate_audio_seconds=max_generate_audio_seconds)
 	return music_generation(args)
 
-
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
 	gr.Markdown("""
     # InspireMusic
@@ -140,11 +144,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 		# with gr.Column():
 		# 	fast = gr.Checkbox(label="Fast Inference", value=False)
 		# 	fade_out = gr.Checkbox(label="Apply Fade Out Effect", value=True)
-	if output_sample_rate == 24000:
-		fast = True
-	else:
-		fast = False
-	fade_out = True
+
 	with gr.Row(equal_height=True):
 		# Textbox for custom input
 		text_input = gr.Textbox(label="Input Text (For Text-to-Music Task)", value="Experience soothing and sensual instrumental jazz with a touch of Bossa Nova, perfect for a relaxing restaurant or spa ambiance.")
@@ -157,7 +157,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 		button = gr.Button("Text to Music")
 		button.click(run_text2music,
 						  inputs=[text_input, model_name,
-								  chorus, fast, fade_out,
+								  chorus,
 								  output_sample_rate,
 								  max_generate_audio_seconds],
 						  outputs=music_output)
@@ -165,7 +165,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 		generate_button = gr.Button("Music Continuation")
 		generate_button.click(run_continuation,
 						  inputs=[text_input, audio_input, model_name,
-								  chorus, fast, fade_out,
+								  chorus,
 								  output_sample_rate,
 								  max_generate_audio_seconds],
 						  outputs=music_output)
@@ -176,7 +176,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 			button = gr.Button(value=prompt)
 			button.click(run_text2music,
 						 inputs=[text_input, model_name,
-								 chorus, fast, fade_out,
+								 chorus,
 								 output_sample_rate,
 								 max_generate_audio_seconds],
 						 outputs=music_output)