Update app.py

app.py

@@ -293,7 +293,7 @@ with gr.Blocks(css=css, theme=theme, fill_width= False) as app:
     with gr.Column(scale=0):
         image_output = gr.Image(show_download_button=True, interactive=False, label='Generated Image 🌄', show_share_button=False, show_fullscreen_button=True, format="png", elem_id="gallery")
         
-        text_button.click(query, inputs=[prompt, negative_prompt, task, steps, sampler, cfg_scale, seed, width, height], outputs=image_output, concurrency_limit=48)
+        text_button.click(query, inputs=[prompt, negative_prompt, style_selection, task, steps, sampler, cfg_scale, seed, width, height], outputs=image_output, concurrency_limit=48)
         clear_prompt =gr.Button("Clear 🗑️",variant="primary", elem_id="clear_button")
         clear_prompt.click(lambda: (None, None), None, [prompt, image_output], queue=False, show_api=False)