Update app.py

app.py

@@ -2,15 +2,36 @@ import os
 import torch
 import torchaudio
 import gradio as gr
+import spaces
 from einops import rearrange
 from stable_audio_3 import StableAudioModel
 
-# Load model once at startup
+# Get token from HF Spaces secrets
+hf_token = os.environ.get("HF_TOKEN")
+
+
+# Required for ZeroGPU Spaces - must have at least one @spaces.GPU function
+# This is a minimal startup check that uses negligible quota
+@spaces.GPU(duration=1)
+def _gpu_startup_check():
+    return "GPU check passed"
+
+
+# Run the check at startup
+_gpu_startup_check()
+
+
+# Load model once at startup (on CPU)
 print("Loading Stable Audio 3 Small model...")
-model = StableAudioModel.from_pretrained("small-music", device="cpu")
+model = StableAudioModel.from_pretrained(
+    "small-music",
+    device="cpu",
+    token=hf_token
+)
 print("Model loaded successfully!")
 
 
+@spaces.GPU(duration=1)
 def generate_audio(prompt, duration, steps, cfg_scale, seed):
     print(f"Generating: prompt='{prompt}', duration={duration}s, steps={steps}, cfg={cfg_scale}, seed={seed}")
 
@@ -28,7 +49,7 @@ def generate_audio(prompt, duration, steps, cfg_scale, seed):
     audio = audio.to(torch.float32).clamp(-1, 1).mul(32767).to(torch.int16).cpu()
 
     output_path = "output.wav"
-    torchaudio.save(output_path, audio, 44100)
+    torchaudio.save(output_path, audio, 44000)
     print("Generation complete!")
     return output_path