Update app.py

app.py

@@ -43,20 +43,30 @@ def generate_audio(prompt, negative_prompt, duration, seed):
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎧 Stable Audio Open - Interactive Audio Generation 🎼")
+    gr.Markdown("## 🎧 Stable Audio Open - Audio Generation 🎼")
+    gr.Markdown("### Adjust prompts, duration, and diffusion steps to control the generation!")
+    
+    # Input Section
     with gr.Row():
         prompt_input = gr.Textbox(label="Prompt", value="The sound of a hammer hitting a wooden surface.")
         negative_input = gr.Textbox(label="Negative Prompt", value="Low quality.")
     with gr.Row():
         duration_input = gr.Slider(minimum=1, maximum=10, step=0.5, value=2, label="Duration (seconds)")
-        seed_input = gr.Number(label="Random Seed", value=0)
+        diffusion_steps_input = gr.Slider(minimum=10, maximum=500, step=10, value=50, label="Diffusion Steps")
+    with gr.Row():
+        seed_input = gr.Number(label="Random Seed", value=42)
+    
+    # Output Section
     generate_button = gr.Button("Generate Audio")
     output_audio = gr.Audio(label="Generated Audio", type="filepath")
     
+    # Connect the function to the button click
     generate_button.click(
         generate_audio, 
-        inputs=[prompt_input, negative_input, duration_input, seed_input], 
+        inputs=[prompt_input, negative_input, duration_input, diffusion_steps_input, seed_input], 
         outputs=output_audio
     )
 
+# Launch the app
 demo.launch()
+