Add a callback called before reloading the server

1 files changed, 1 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index 5f7a53f9..dda34249 100644
--- a/webui.py
+++ b/webui.py
@@ -108,6 +108,7 @@ def wait_on_server(demo=None):
         time.sleep(0.5)

         modules.script_callbacks.app_polling_callback(None, demo)

         if shared.state.need_restart:

+            modules.script_callbacks.app_reload_callback(None, demo)

             shared.state.need_restart = False

             time.sleep(0.5)

             demo.close()