Merge pull request #1 from mouhao/mouhao-patch-1

Update webui.py

1 files changed, 8 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index 357bf4c1..a3dfdcee 100644
--- a/webui.py
+++ b/webui.py
@@ -286,6 +286,11 @@ def api_only():
     print(f"Startup time: {startup_timer.summary()}.")

     api.launch(server_name="0.0.0.0" if cmd_opts.listen else "127.0.0.1", port=cmd_opts.port if cmd_opts.port else 7861)

 

+# patch in url for api docs

+def my_setup(self):

+    self.docs_url = "/docs"

+    self.redoc_url = "/redoc"

+    self.orig_setup()

 

 def webui():

     launch_api = cmd_opts.api

@@ -313,6 +318,9 @@ def webui():
                 for line in file.readlines():

                     gradio_auth_creds += [x.strip() for x in line.split(',') if x.strip()]

 

+        if launch_api:

+            FastAPI.orig_setup = FastAPI.setup

+            setattr(FastAPI, "setup", my_setup)            

         app, local_url, share_url = shared.demo.launch(

             share=cmd_opts.share,

             server_name=server_name,