Update app.py

app.py

@@ -29,7 +29,12 @@ os.makedirs(OUTPUT_PATH, exist_ok=True)
 def generate_audio(prompt: str):
     # Generate the audio using StableAudioPipeline
     generator = torch.Generator(device).manual_seed(42)
-    audio_output = pipe(prompt=prompt, negative_prompt="Low Quality", num_inference_steps=10).audios
+    audio_output = pipe(prompt=prompt,
+        negative_prompt='Low Quality',
+        num_inference_steps=int(10),  # Number of diffusion steps
+        audio_end_in_s=1,
+        num_waveforms_per_prompt=1,
+        generator=generator).audios
     
     # Convert to numpy and save to a WAV file
     output_audio = audio_output[0].T.float().cpu().numpy()