Update app.py

app.py

@@ -118,7 +118,7 @@ with gr.Blocks() as demo:
     output_images = gr.Gallery(label="Generated Images", elem_id="gallery", show_label=False)
     
     # Connecting the button click to the image generation function
-    run_button.click(infer_gradio, inputs=[prompt_input, output_images], outputs=output_images)
+    run_button.click(infer_gradio, inputs=[prompt_input], outputs=output_images)
 
 # Launch Gradio app
 demo.launch()