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,