update

app.py

@@ -187,13 +187,14 @@ def main():
 	demo.launch()
 
 if __name__ == '__main__':
-	model_list = ["InspireMusic-Base", "InspireMusic-1.5B-Long"]
-
+	# model_list = ["InspireMusic-Base", "InspireMusic-1.5B-Long"]
+	model_list = ["InspireMusic-Base"]
 	for model_name in model_list:
 		model_dir = f"pretrained_models/{model_name}"
 		if not os.path.isdir(model_dir):
 			if model_name == "InspireMusic-Base":
-				snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
+				local_model_dir = snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
 			else:
-				snapshot_download(f"iic/{model_name}", local_dir=model_dir)
+				local_model_dir = snapshot_download(f"iic/{model_name}", local_dir=model_dir)
+			print(local_model_dir, model_dir)
 	main()