update

app.py

@@ -12,12 +12,7 @@ os.system('nvidia-smi')
 print(torch.backends.cudnn.version())
 
 def generate_filename():
-	now = datetime.datetime.now()
-	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_object = hashlib.sha256(str(int(datetime.datetime.now().timestamp())).encode())
 	hash_string = hash_object.hexdigest()
 	return hash_string
 
@@ -60,6 +55,15 @@ def get_args(
 	print(args)
 	return args
 
+
+def trim_audio(audio_file, cut_seconds=5):
+	audio, sr = torchaudio.load(audio_file)
+	num_samples = cut_seconds * sr
+	cutted_audio = audio[:, :num_samples]
+	output_path = os.path.join(os.getcwd(), "audio_prompt_" + generate_filename() + ".wav")
+	torchaudio.save(output_path, cutted_audio, sr)
+	return output_path
+
 @spaces.GPU
 def music_generation(args):
 	set_env_variables()
@@ -90,24 +94,6 @@ def music_generation(args):
 			trim=args["trim"])
 	return output_path
 
-def update_text():
-    global text_input  # Declare as global to modify the outer scope variable
-    text_input = "New value set by button click"
-    return text_input
-
-default_prompts = [
-    "Experience soothing and sensual instrumental jazz with a touch of Bossa Nova, perfect for a relaxing restaurant or spa ambiance.",
-	"Compose an uplifting R&B song.",
-	"Create an emotional, introspective folk song with acoustic guitar and soft vocals."
-]
-
-def trim_audio(audio_file, cut_seconds=5):
-	audio, sr = torchaudio.load(audio_file)
-	num_samples = cut_seconds * sr
-	cutted_audio = audio[:, :num_samples]
-	output_path = os.path.join(os.getcwd(), "audio_prompt_" + generate_filename() + ".wav")
-	torchaudio.save(output_path, cutted_audio, sr)
-	return output_path
 
 @spaces.GPU
 def demo_inspiremusic_t2m(text, model_name, chorus,
@@ -151,7 +137,6 @@ def main():
 												   value=30)
 
 		with gr.Row(equal_height=True):
-			# Textbox for custom input
 			text_input = gr.Textbox(label="Input Text (For Text-to-Music Task)", value="Experience soothing and sensual instrumental jazz with a touch of Bossa Nova, perfect for a relaxing restaurant or spa ambiance.")
 
 			audio_input = gr.Audio(label="Input Audio Prompt (For Music Continuation Task)",
@@ -177,8 +162,8 @@ def main():
 	demo.launch()
 
 if __name__ == '__main__':
-	model_list = ["InspireMusic-Base", "InspireMusic-1.5B-Long"]
-	# model_list = ["InspireMusic-Base"]
+	model_list = ["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-Base"]
+
 	for model_name in model_list:
 		model_dir = f"pretrained_models/{model_name}"
 		if not os.path.isdir(model_dir):