update

app.py

@@ -136,33 +136,34 @@ def main():
 												   label="Generate Audio Length (s)",
 												   value=30)
 
-		with gr.Row(equal_height=True):
-			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.")
-
-			audio_input = gr.Audio(label="Input Audio Prompt (For Music Continuation Task)",
+		# with gr.Row(equal_height=True):
+		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.")
+		music_output = gr.Audio(label="Text to Music Output", type="filepath")
+
+		button = gr.Button("Text to Music")
+		button.click(demo_inspiremusic_t2m,
+						  inputs=[text_input, model_name,
+								  chorus,
+								  output_sample_rate,
+								  max_generate_audio_seconds],
+						  outputs=music_output)
+
+		audio_input = gr.Audio(label="Input Audio Prompt (For Music Continuation Task)",
 								   type="filepath")
-		music_output = gr.Audio(label="Generated Music", type="filepath")
-
-		with gr.Row():
-			button = gr.Button("Text to Music")
-			button.click(demo_inspiremusic_t2m,
-							  inputs=[text_input, model_name,
-									  chorus,
-									  output_sample_rate,
-									  max_generate_audio_seconds],
-							  outputs=music_output)
 
-			generate_button = gr.Button("Music Continuation")
-			generate_button.click(demo_inspiremusic_con,
+		music_con_output = gr.Audio(label="Music Continuation Output", type="filepath")
+		generate_button = gr.Button("Music Continuation")
+		generate_button.click(demo_inspiremusic_con,
 							  inputs=[text_input, audio_input, model_name,
 									  chorus,
 									  output_sample_rate,
 									  max_generate_audio_seconds],
-							  outputs=music_output)
+							  outputs=music_con_output)
 	demo.launch()
 
 if __name__ == '__main__':
-	model_list = ["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-Base"]
+	# model_list = ["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-Base"]
+	model_list = ["InspireMusic-Base"]
 
 	for model_name in model_list:
 		model_dir = f"pretrained_models/{model_name}"