Update app.py

app.py

@@ -56,20 +56,18 @@ def startup():
 # -------- API --------
 @app.post("/generate")
 async def generate(prompt: str, duration: float = 10.0):
-    """
-    Generate music/audio from a text prompt using Stable Audio Open
-    """
     load_pipeline()
 
+    # ✅ Stable Audio Open expects duration as a pipeline attribute
+    pipe.audio_length_in_s = float(duration)
+
     with torch.no_grad():
         output = pipe(
             prompt=prompt,
-            audio_length_in_s=duration,
             guidance_scale=7.5,
             num_inference_steps=150,
         )
 
-
     audio = output.audios[0]
 
     filename = f"{uuid.uuid4().hex}.wav"
@@ -82,3 +80,4 @@ async def generate(prompt: str, duration: float = 10.0):
         "duration": duration,
         "file": filename,
     }
+