update

app.py

@@ -7,6 +7,8 @@ import datetime
 import hashlib
 import torch
 from modelscope import snapshot_download
+
+os.system('nvidia-smi')
 print(torch.backends.cudnn.version())
 
 def generate_filename():
@@ -185,10 +187,10 @@ def main():
 	demo.launch()
 
 if __name__ == '__main__':
-	model_list = ["InspireMusic-Base"]
+	model_list = ["InspireMusic-Base", "InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-Base-24kHz", "InspireMusic-1.5B-Long-24kHz"]
 	for model_name in model_list:
 		if model_name == "InspireMusic-Base":
 			snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
 		else:
-			snapshot_download(f"iic/{model_name}", local_dir=model_dir)os.system('nvidia-smi')
+			snapshot_download(f"iic/{model_name}", local_dir=model_dir)
 	main()