update

app.py

@@ -13,15 +13,6 @@
 # limitations under the License.
 
 import os
-import spaces
-import gradio as gr
-from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
-import torchaudio
-import datetime
-import hashlib
-import torch
-import importlib
-import sys
 
 os.system('nvidia-smi')
 os.system('apt update -y && apt-get install -y apt-utils && apt install -y unzip')
@@ -32,12 +23,23 @@ os.environ['PYTHONPATH'] = 'third_party/Matcha-TTS'
 os.system('mkdir pretrained_models && cd pretrained_models && git clone https://huggingface.co/FunAudioLLM/InspireMusic-Base.git &&git clone https://huggingface.co/FunAudioLLM/InspireMusic-1.5B-Long.git &&git clone https://huggingface.co/FunAudioLLM/InspireMusic-1.5B.git &&git clone https://huggingface.co/FunAudioLLM/InspireMusic-1.5B-24kHz.git &&git clone https://huggingface.co/FunAudioLLM/InspireMusic-Base-24kHz.git && for i in InspireMusic-Base InspireMusic-Base-24kHz InspireMusic-1.5B InspireMusic-1.5B-24kHz InspireMusic-1.5B-Long; do sed -i -e "s/\.\.\/\.\.\///g" ${i}/inspiremusic.yaml; done && cd ..')
 
 # os.system('mkdir pretrained_models && cd pretrained_models && git clone https://huggingface.co/FunAudioLLM/InspireMusic-Base.git && for i in InspireMusic-Base; do sed -i -e "s/\.\.\/\.\.\///g" ${i}/inspiremusic.yaml; done && cd ..')
-
+import sys
+import torch
 print(torch.backends.cudnn.version())
 
 ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
 sys.path.append('{}/third_party/Matcha-TTS'.format(ROOT_DIR))
 
+import spaces
+import gradio as gr
+from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
+import torchaudio
+import datetime
+import hashlib
+import importlib
+
+MODELS = ["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-1.5B-24kHz", "InspireMusic-Base", "InspireMusic-Base-24kHz"]
+
 def generate_filename():
 	hash_object = hashlib.sha256(str(int(datetime.datetime.now().timestamp())).encode())
 	hash_string = hash_object.hexdigest()
@@ -154,7 +156,7 @@ def main():
 		""")
 
 		with gr.Row(equal_height=True):
-			model_name = gr.Dropdown(["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-1.5B-24kHz", "InspireMusic-Base", "InspireMusic-Base-24kHz"], label="Select Model Name", value="InspireMusic-1.5B-Long")
+			model_name = gr.Dropdown(MODELS, label="Select Model Name", value="InspireMusic-1.5B-Long")
 			chorus = gr.Dropdown(["intro", "verse", "chorus", "outro"],
 								 label="Chorus Mode", value="intro")
 			output_sample_rate = gr.Dropdown([48000, 24000],