Update app.py

app.py

@@ -17,19 +17,29 @@ DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 pipe = None  # lazy-loaded
 
 
+from huggingface_hub import login
+
 def load_pipeline():
     global pipe
     if pipe is None:
         print("Loading Stable Audio Open pipeline...")
+
+        # Authenticate inside Space
+        token = os.environ.get("HF_TOKEN")
+        if token:
+            login(token=token)
+
         pipe = DiffusionPipeline.from_pretrained(
             "stabilityai/stable-audio-open-1.0",
             torch_dtype=DTYPE,
+            use_auth_token=True,  # IMPORTANT
         )
         pipe.to(DEVICE)
         pipe.set_progress_bar_config(disable=True)
         print("Pipeline loaded on", DEVICE)
 
 
+
 # -------- API --------
 @app.post("/generate")
 async def generate(prompt: str, duration: float = 10.0):