michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[Feature Request] Please add "--share-password" for Gradio server security #315

Browse source
This commit is contained in:
AUTOMATIC 2022-09-12 15:10:05 +03:00
parent e4f080f61a
commit 372a2c3e2e
2 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/shared.py
View file

@ -42,7 +42,9 @@ parser.add_argument("--show-negative-prompt", action='store_true', help="does no
parser.add_argument("--ui-config-file", type=str, help="filename to use for ui configuration", default=os.path.join(script_path, 'ui-config.json'))
parser.add_argument("--hide-ui-dir-config", action='store_true', help="hide directory configuration from webui", default=False)
parser.add_argument("--ui-settings-file", type=str, help="filename to use for ui settings", default=os.path.join(script_path, 'config.json'))
parser.add_argument("--gradio-debug", action='store_true', help="launch gradio with --debug option")
parser.add_argument("--gradio-debug", action='store_true', help="launch gradio with --debug option")
parser.add_argument("--gradio-username", type=str, help="username for gradio authentication; default: user", default="user")
parser.add_argument("--gradio-password", type=str, help="password for gradio authentication; default: no authentication", default=None)
cmd_opts = parser.parse_args()

8
webui.py
View file

@ -115,7 +115,13 @@ def webui():
run_pnginfo=modules.extras.run_pnginfo
)
demo.launch(share=cmd_opts.share, server_name="0.0.0.0" if cmd_opts.listen else None, server_port=cmd_opts.port, debug=cmd_opts.gradio_debug)
demo.launch(
share=cmd_opts.share,
server_name="0.0.0.0" if cmd_opts.listen else None,
server_port=cmd_opts.port,
debug=cmd_opts.gradio_debug,
auth=(cmd_opts.gradio_username, cmd_opts.gradio_password) if cmd_opts.gradio_password is not None else None,
)
if __name__ == "__main__":