Update app.py

app.py

@@ -126,7 +126,7 @@ def create_ui():
             output_image = gr.Image(label="Generated Image")
         
         # Link the button to the function
-        generate_button.click(fn=generate_image, inputs=prompt_input, outputs=output_image)
+        generate_button.click(fn=generate_image, inputs=[prompt_input, True], outputs=output_image)
     
     return ui