diff --git a/scripts/inifnite-zoom.py b/scripts/inifnite-zoom.py index fe3053e..9353ec0 100644 --- a/scripts/inifnite-zoom.py +++ b/scripts/inifnite-zoom.py @@ -250,6 +250,7 @@ def on_ui_tabs(): """ ) generate_btn = gr.Button(value="Generate video", variant="primary") + interrupt = gr.Button(value="Interrupt", elem_id="interrupt_training") with gr.Row(): with gr.Column(scale=1, variant="panel"): @@ -381,7 +382,11 @@ def on_ui_tabs(): ], outputs=[output_video, out_image, generation_info, html_info, html_log], ) - + interrupt.click( + fn=lambda: shared.state.interrupt(), + inputs=[], + outputs=[] + ) return [(infinite_zoom_interface, "Infinite Zoom", "iz_interface")] def on_ui_settings():