Update app.py

app.py

@@ -191,6 +191,6 @@ with gr.Blocks(theme='gstaff/xkcd', css=css) as app:
         with gr.Row():
             image_output = gr.Image(type="pil", label="Image Output", elem_id="gallery")
         
-        text_button.click(query, inputs=[text_prompt, is_realistic, num_inference_steps, width, height], outputs=image_output, mod=mod_choice)
+        text_button.click(query, inputs=[text_prompt, is_realistic, num_inference_steps, width, height, mod_choice], outputs=image_output)
 
 app.launch(show_api=True, share=True)