Update app.py

app.py

@@ -157,6 +157,6 @@ with gr.Blocks(css=css) as block:
         gr.HTML(article)
 
     train_btn.click(fn=train, inputs=[prompt_input,image_init,trn_text,trn_steps], outputs=[training_status])
-    gen_btn.click(fn=generate, inputs=[prompt_input,image_init], outputs=[image_output])
+    gen_btn.click(fn=generate, inputs=[prompt_input,image_init,trn_text,trn_steps], outputs=[image_output])
     
 block.queue(max_size=12).launch(show_api=False)