Update app.py

app.py

@@ -40,3 +40,23 @@ def generate_audio(prompt, negative_prompt, duration, seed):
     output_audio = audio_output[0].T.float().cpu().numpy()
     sf.write("output.wav", output_audio, pipe.vae.sampling_rate)
     return "output.wav"
+
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 🎧 Stable Audio Open - Interactive Audio Generation 🎼")
+    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)
+    generate_button = gr.Button("Generate Audio")
+    output_audio = gr.Audio(label="Generated Audio", type="filepath")
+    
+    generate_button.click(
+        generate_audio, 
+        inputs=[prompt_input, negative_input, duration_input, seed_input], 
+        outputs=output_audio
+    )
+
+demo.launch()