Update app.py

app.py

@@ -268,21 +268,21 @@ with block:
                 result_gallery = gr.Gallery(height=512, object_fit='contain', label='Outputs', columns=4)
 
     input_fg.change(lambda: ["", gr.update(interactive=True), gr.update(interactive=False), gr.update(interactive=False)],
-                    outputs=[prompt, prompt_gen_button, key_gen_button, i2v_end_btn])
+                    outputs=[prompt, prompt_gen_button, key_gen_button])
 
     prompt_gen_button.click(
         fn=interrogator_process,
         inputs=[input_fg],
         outputs=[prompt, i2v_input_text]
     ).then(lambda: [gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=False)],
-           outputs=[prompt_gen_button, key_gen_button, i2v_end_btn])
+           outputs=[prompt_gen_button, key_gen_button])
 
     key_gen_button.click(
         fn=process,
         inputs=[input_fg, prompt, input_undo_steps, image_width, image_height, seed, steps, n_prompt, cfg],
         outputs=[result_gallery]
     ).then(lambda: [gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)],
-           outputs=[prompt_gen_button, key_gen_button, i2v_end_btn])
+           outputs=[prompt_gen_button, key_gen_button])
 
 
     dbs = [