1 files changed, 2 insertions, 0 deletions
diff --git a/webui.py b/webui.py
index ae3285c6..98b6d394 100644
--- a/webui.py
+++ b/webui.py
@@ -264,7 +264,9 @@ def create_api(app):
 def wait_on_server(demo=None):
     while 1:
         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()