update

app.py

@@ -1,6 +1,33 @@
 import os
+import sys
 import gradio as gr
 from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
+from modelscope import snapshot_download
+# /home/user/app
+
+import datetime
+import hashlib
+
+def generate_filename():
+	# Get the current date-time
+	now = datetime.datetime.now()
+	# Get total seconds since the epoch
+	seconds_since_epoch = int(now.timestamp())
+	# Convert seconds to string
+	seconds_str = str(seconds_since_epoch)
+	# Hash the string using SHA-256
+	hash_object = hashlib.sha256(seconds_str.encode())
+	hash_string = hash_object.hexdigest()
+	return hash_string
+
+def download_model(model_name="InspireMusic-Base"):
+	# model_dir = "pretrained_models"
+	model_dir = os.path.join(os.getcwd(), f"iic/{model_name}")
+	if not os.path.exists(model_dir):
+		model_dir = snapshot_download(f"iic/{model_name}")
+	print(model_dir)
+	return model_dir
+
 
 def get_args(
 		task, text="", audio=None, model_name="InspireMusic-Base",
@@ -18,12 +45,13 @@ def get_args(
 		"trim"                      : trim,
 		"output_sample_rate"        : output_sample_rate,
 		"max_generate_audio_seconds": max_generate_audio_seconds,
-		"model_dir"                 : os.path.join("pretrained_models",
+		"model_dir"                 : os.path.join("iic",
 												   model_name),
 		"result_dir"                : "exp/inspiremusic",
-		"output_fn"                 : "output_audio",
+		"output_fn"                 : generate_filename(),
 		"format"                    : "wav",
 	}
+	print(args)
 	return args
 
 

requirements.txt

@@ -2,7 +2,7 @@
 conformer==0.3.2
 diffusers==0.27.2
 gdown==5.1.0
-gradio
+gradio==4.32.2
 grpcio==1.57.0
 grpcio-tools==1.57.0
 hydra-core==1.3.2
@@ -27,3 +27,4 @@ WeTextProcessing==1.0.3
 transformers
 accelerate
 huggingface-hub==0.25.2
+hashlib