From 5427b7128db9656038bebc0a0240ee122b98e6b9 Mon Sep 17 00:00:00 2001 From: mouhao Date: Sun, 7 May 2023 20:54:48 +0800 Subject: Update webui.py Fix missing /docs endpoint in newer gradio versions Newer versions of gradio (>=3.27.1) have removed the /docs endpoint by default. This commit adds it back to enable accessing the API documentation. --- webui.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'webui.py') 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, -- cgit v1.2.1