Merge pull request #4155 from MaikoTan/fix/register-api-in-api-only-mode

fix: should invoke callback as well in api only mode

1 files changed, 2 insertions, 1 deletions

diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py
index da88635b..c28e220e 100644
--- a/modules/script_callbacks.py
+++ b/modules/script_callbacks.py
@@ -2,6 +2,7 @@ import sys
 import traceback

 from collections import namedtuple

 import inspect

+from typing import Optional

 

 from fastapi import FastAPI

 from gradio import Blocks

@@ -62,7 +63,7 @@ def clear_callbacks():
     callbacks_image_saved.clear()

     callbacks_cfg_denoiser.clear()

 

-def app_started_callback(demo: Blocks, app: FastAPI):

+def app_started_callback(demo: Optional[Blocks], app: FastAPI):

     for c in callbacks_app_started:

         try:

             c.callback(demo, app)