modified: app.py

app.py

@@ -19,6 +19,7 @@ from torchaudio.transforms import Resample
 from einops import rearrange
 from tqdm import tqdm
 from omegaconf import OmegaConf
+import spaces
 
 # --- Constants and Environment Setup ---
 IS_SHARED_UI = "innova-ai/YuE-music-generator-demo" in os.environ.get('SPACE_ID', '')
@@ -165,6 +166,7 @@ class BlockTokenRangeProcessor(LogitsProcessor):
         return scores
 
 # --- Music Generation Core Function ---
+@spaces.GPU(duration=120)
 def generate_music(
     stage1_model,
     mmtokenizer,