update

app.py

@@ -38,7 +38,7 @@ import datetime
 import hashlib
 import importlib
 
-MODELS = ["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-1.5B-24kHz", "InspireMusic-Base", "InspireMusic-Base-24kHz"]
+MODELS = ["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-Base", "InspireMusic-1.5B-24kHz", "InspireMusic-Base-24kHz"]
 AUDIO_PROMPT_DIR = "demo/audio_prompts"
 OUTPUT_AUDIO_DIR = "demo/outputs"
 
@@ -171,6 +171,8 @@ def main():
 			max_generate_audio_seconds = gr.Slider(10, 300,
 												   label="Generate Audio Length (s)",
 												   value=30)
+		if "24kHz" in model_name:
+			output_sample_rate = 24000
 
 		with gr.Row(equal_height=True):
 			text_input = gr.Textbox(label="Input Text (For Text-to-Music Task)",