update

app.py

@@ -5,10 +5,23 @@ import gradio as gr
 import numpy as np
 import torch
 
+try:
+    import spaces
+except ImportError:
+    class _SpacesFallback:
+        @staticmethod
+        def GPU(*_args, **_kwargs):
+            def _decorator(func):
+                return func
+
+            return _decorator
+
+    spaces = _SpacesFallback()
+
 from moss_soundeffect_v2 import MossSoundEffectPipeline
 
 MODEL_PATH = "OpenMOSS-Team/MOSS-SoundEffect-v2.0"
-DEFAULT_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+DEFAULT_DEVICE = "cuda"
 MAX_INFERENCE_SECONDS = 30
 
 
@@ -23,6 +36,7 @@ def load_backend(model_dir: str, device_str: str):
     return pipe, device
 
 
+@spaces.GPU(duration=180)
 def run_inference(prompt, seconds, steps, cfg_scale, sigma_shift, seed):
     if not (prompt or "").strip():
         raise ValueError("Please enter a prompt describing the audio you want to generate.")

requirements.txt

@@ -1,4 +1,5 @@
 gradio==6.11.0
+spaces
 numpy==1.26.4
 einops==0.8.2
 pillow==12.2.0