Fix ZeroGPU compatibility and auth

app.py

@@ -4,10 +4,13 @@ import torchaudio
 import gradio as gr
 import spaces
 from einops import rearrange
+from huggingface_hub import login
 from stable_audio_3 import StableAudioModel
 
-# Get token from HF Spaces secrets
+# Authenticate with gated model when HF_TOKEN secret is present
 hf_token = os.environ.get("HF_TOKEN")
+if hf_token:
+    login(token=hf_token)
 
 
 # Required for ZeroGPU Spaces - must have at least one @spaces.GPU function
@@ -25,8 +28,7 @@ _gpu_startup_check()
 print("Loading Stable Audio 3 Small model...")
 model = StableAudioModel.from_pretrained(
     "small-music",
-    device="cpu",
-    token=hf_token
+    device="cpu"
 )
 print("Model loaded successfully!")
 

requirements.txt

@@ -1,7 +1,7 @@
-git+https://github.com/Stability-AI/stable-audio-3.git
 gradio==6.3.0
 soundfile>=0.13.1
 einops
 torch>=2.7.1
 torchaudio>=2.7.1
 huggingface-hub>=1.7.1
+git+https://github.com/Stability-AI/stable-audio-3.git