Disable api

app_flux.py

@@ -1,5 +1,5 @@
-import os
 import replicate_gradio
+
 from utils import get_app
 
 demo = get_app(
@@ -7,11 +7,10 @@ demo = get_app(
         "black-forest-labs/flux-depth-pro",
         "black-forest-labs/flux-canny-pro",
         "black-forest-labs/flux-fill-pro",
-        "black-forest-labs/flux-depth-dev"
+        "black-forest-labs/flux-depth-dev",
     ],
     default_model="black-forest-labs/flux-depth-pro",
     src=replicate_gradio.registry,
-    is_chat=False
 )
 
 if __name__ == "__main__":

app_ltx_video.py

@@ -1,5 +1,5 @@
-import os
 import fal_gradio
+
 from utils import get_app
 
 demo = get_app(
@@ -9,7 +9,6 @@ demo = get_app(
     ],
     default_model="fal-ai/ltx-video/image-to-video",
     src=fal_gradio.registry,
-    is_chat=False
 )
 
 if __name__ == "__main__":

utils.py

@@ -8,7 +8,6 @@ def get_app(
     default_model: str,
     src: Callable[[str, str | None], gr.Blocks] | Literal["models"],
     accept_token: bool = False,
-    is_chat: bool = True,
     **kwargs,
 ) -> gr.Blocks:
     def update_model(new_model: str) -> list[gr.Column]:
@@ -32,12 +31,7 @@ def get_app(
             queue=False,
         )
 
-    if not is_chat:
-        for k in list(demo.fns.keys()):
-            del demo.fns[k]
-    else:
-        for k, v in list(demo.fns.items()):
-            if isinstance(v.api_name, str) and "chat" in v.api_name:
-                del demo.fns[k]
+    for fn in demo.fns.values():
+        fn.api_name = False
 
     return demo