diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-01 17:33:31 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-01 17:33:31 +0000 |
| commit | 4f8490cd5630823ac44de8b5c5e4325bdbbea7fa (patch) | |
| tree | 98a7cffafdf6b5e52e72151c8829d03589b76f50 /modules/ui.py | |
| parent | 95f35d04ab1636e08f69ca9c0ae2446714870e80 (diff) | |
| download | stable-diffusion-webui-gfx803-4f8490cd5630823ac44de8b5c5e4325bdbbea7fa.tar.gz stable-diffusion-webui-gfx803-4f8490cd5630823ac44de8b5c5e4325bdbbea7fa.tar.bz2 stable-diffusion-webui-gfx803-4f8490cd5630823ac44de8b5c5e4325bdbbea7fa.zip | |
add restart button
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/modules/ui.py b/modules/ui.py index 15572bb0..ec6aaa28 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1002,6 +1002,17 @@ def create_ui(txt2img, img2img, run_extras, run_pnginfo, run_modelmerger): _js='function(){}'
)
+ def request_restart():
+ settings_interface.gradio_ref.do_restart = True
+
+ restart_gradio = gr.Button(value='Restart Gradio and Refresh Scripts')
+ restart_gradio.click(
+ fn=request_restart,
+ inputs=[],
+ outputs=[],
+ _js='function(){document.body.innerHTML=\'<h1 style="color:white;text-align: center;">Reloading</h1>\';setTimeout(function(){location.reload()},2000)}'
+ )
+
if column is not None:
column.__exit__()
@@ -1026,7 +1037,9 @@ def create_ui(txt2img, img2img, run_extras, run_pnginfo, run_modelmerger): css += css_hide_progressbar
with gr.Blocks(css=css, analytics_enabled=False, title="Stable Diffusion") as demo:
-
+
+ settings_interface.gradio_ref = demo
+
with gr.Tabs() as tabs:
for interface, label, ifid in interfaces:
with gr.TabItem(label, id=ifid):
|