From 9d669572075584be3a155a7fd40dbf79a382e64b Mon Sep 17 00:00:00 2001 From: v0xie <28695009+v0xie@users.noreply.github.com> Date: Thu, 13 Apr 2023 17:35:08 -0700 Subject: [PATCH] Add --listen-host launch option (#1122) --- README.md | 1 + modules/shared.py | 1 + server.py | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 257af117..76680297 100644 --- a/README.md +++ b/README.md @@ -265,6 +265,7 @@ Optionally, you can use the following command-line flags: | Flag | Description | |---------------------------------------|-------------| | `--listen` | Make the web UI reachable from your local network. | +| `--listen-host LISTEN_HOST` | The hostname that the server will use. | | `--listen-port LISTEN_PORT` | The listening port that the server will use. | | `--share` | Create a public URL. This is useful for running the web UI on Google Colab or similar. | | `--auto-launch` | Open the web UI in the default browser upon launch. | diff --git a/modules/shared.py b/modules/shared.py index 41ca3132..ed9c2c02 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -136,6 +136,7 @@ parser.add_argument('--rwkv-cuda-on', action='store_true', help='RWKV: Compile t # Gradio parser.add_argument('--listen', action='store_true', help='Make the web UI reachable from your local network.') +parser.add_argument('--listen-host', type=str, help='The hostname that the server will use.') parser.add_argument('--listen-port', type=int, help='The listening port that the server will use.') parser.add_argument('--share', action='store_true', help='Create a public URL. This is useful for running the web UI on Google Colab or similar.') parser.add_argument('--auto-launch', action='store_true', default=False, help='Open the web UI in the default browser upon launch.') diff --git a/server.py b/server.py index 8feb0819..15af8373 100644 --- a/server.py +++ b/server.py @@ -819,7 +819,7 @@ def create_interface(): # Launch the interface shared.gradio['interface'].queue() if shared.args.listen: - shared.gradio['interface'].launch(prevent_thread_lock=True, share=shared.args.share, server_name='0.0.0.0', server_port=shared.args.listen_port, inbrowser=shared.args.auto_launch, auth=auth) + shared.gradio['interface'].launch(prevent_thread_lock=True, share=shared.args.share, server_name=shared.args.listen_host or '0.0.0.0', server_port=shared.args.listen_port, inbrowser=shared.args.auto_launch, auth=auth) else: shared.gradio['interface'].launch(prevent_thread_lock=True, share=shared.args.share, server_port=shared.args.listen_port, inbrowser=shared.args.auto_launch, auth=auth)