Update app.py

app.py

@@ -23,7 +23,7 @@ pipe = StableAudioPipeline.from_pretrained(
 pipe = pipe.to(device)
 
 # Function to generate audio
-def generate_audio(prompt, negative_prompt='Low Quality', duration, diffusion_steps, seed):
+def generate_audio(prompt, negative_prompt, duration, diffusion_steps, seed):
     generator = torch.Generator(device).manual_seed(seed)
     audio_output = pipe(
         prompt=prompt,
@@ -46,6 +46,7 @@ with gr.Blocks() as demo:
     # 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=1, label="Duration (seconds)")
         diffusion_steps_input = gr.Slider(minimum=1, maximum=500, step=10, value=10, label="Diffusion Steps")
@@ -59,7 +60,7 @@ with gr.Blocks() as demo:
     # Connect the function to the button click
     generate_button.click(
         generate_audio, 
-        inputs=[prompt_input, duration_input, diffusion_steps_input, seed_input], 
+        inputs=[prompt_input, negative_input, duration_input, diffusion_steps_input, seed_input], 
         outputs=output_audio
     )