From a7333f4023dace20e492e8c2912b82a79b53eabd Mon Sep 17 00:00:00 2001 From: nicholassun99 Date: Sat, 15 Apr 2023 15:13:55 +0800 Subject: [PATCH] Added Interrupt Button Signed-off-by: nicholassun99 --- scripts/inifnite-zoom.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/scripts/inifnite-zoom.py b/scripts/inifnite-zoom.py index 17b7b02..7c576c4 100644 --- a/scripts/inifnite-zoom.py +++ b/scripts/inifnite-zoom.py @@ -248,6 +248,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"): with gr.Tab("Main"): @@ -376,7 +377,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")]