update

app.py

@@ -20,12 +20,9 @@ import torchaudio
 import datetime
 import hashlib
 import torch
-import subprocess
 import importlib
 import sys
 
-
-
 os.system('nvidia-smi')
 os.system('apt update -y && apt-get install -y apt-utils && apt install -y unzip')
 os.system('pip install flash-attn --no-build-isolation')
@@ -92,7 +89,7 @@ def trim_audio(audio_file, cut_seconds=5):
 	torchaudio.save(output_path, cutted_audio, sr)
 	return output_path
 
-# @spaces.GPU
+@spaces.GPU(duration=600)
 def music_generation(args):
 	set_env_variables()
 	model = InspireMusicUnified(
@@ -123,7 +120,7 @@ def music_generation(args):
 	return output_path
 
 
-# @spaces.GPU
+@spaces.GPU
 def demo_inspiremusic_t2m(text, model_name, chorus,
 					 output_sample_rate, max_generate_audio_seconds):
 	args = get_args(
@@ -133,7 +130,7 @@ def demo_inspiremusic_t2m(text, model_name, chorus,
 			max_generate_audio_seconds=max_generate_audio_seconds)
 	return music_generation(args)
 
-# @spaces.GPU
+@spaces.GPU
 def demo_inspiremusic_con(text, audio, model_name, chorus,
 					 output_sample_rate, max_generate_audio_seconds):
 	args = get_args(
@@ -143,7 +140,7 @@ def demo_inspiremusic_con(text, audio, model_name, chorus,
 			max_generate_audio_seconds=max_generate_audio_seconds)
 	return music_generation(args)
 
-# @spaces.GPU
+@spaces.GPU(duration=6000)
 def main():
 	with gr.Blocks(theme=gr.themes.Soft()) as demo:
 		gr.Markdown("""
@@ -192,5 +189,4 @@ def main():
 
 if __name__ == '__main__':
 	# subprocess.run(['bash', 'install.sh'], shell=True)
-	@spaces.GPU
 	main()