From eace07043eb2537c100c01134d2e7f5699c8863b Mon Sep 17 00:00:00 2001 From: venshine Date: Thu, 29 Jun 2023 20:35:36 +0800 Subject: [PATCH] Support configuring gradio startup parameters --- README.md | 4 ++-- visualizer_drag_gradio.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 2c923dc..f8cd72d 100644 --- a/README.md +++ b/README.md @@ -92,9 +92,9 @@ You can run DragGAN Gradio demo as well: ```sh python visualizer_drag_gradio.py ``` -If you want to run DragGAN Gradio demo with 0.0.0.0 as server name, allowing to respond to network requests: +If you want to run DragGAN Gradio demo with given server name, for example host 0.0.0.0 to accessible on local network: ```sh -python visualizer_drag_gradio.py --listen +python visualizer_drag_gradio.py --host=0.0.0.0 ``` If you want to run DragGAN Gradio demo with given server port, for example port 8888: ```sh diff --git a/visualizer_drag_gradio.py b/visualizer_drag_gradio.py index 1f27ab8..418fe81 100644 --- a/visualizer_drag_gradio.py +++ b/visualizer_drag_gradio.py @@ -18,8 +18,8 @@ import inspect parser = ArgumentParser() parser.add_argument('--share', action='store_true',default='True') parser.add_argument('--cache-dir', type=str, default='./checkpoints') -parser.add_argument('--listen', action='store_true', - help="launch gradio with 0.0.0.0 as server name, allowing to respond to network requests") +parser.add_argument('--host', type=str, + help="launch gradio with given server name", default=None) parser.add_argument('--port', type=int, help="launch gradio with given server port", default=None) args = parser.parse_args() @@ -873,4 +873,4 @@ def_server_port = params["server_port"].default gr.close_all() app.queue(concurrency_count=3, max_size=20) -app.launch(share=args.share, server_name="0.0.0.0" if args.listen else def_server_name, server_port=args.port if args.port is not None else def_server_port) +app.launch(share=args.share, server_name=args.host if args.host is not None else def_server_name, server_port=args.port if args.port is not None else def_server_port)