diff --git a/README.md b/README.md index b26a764..2c923dc 100644 --- a/README.md +++ b/README.md @@ -92,6 +92,14 @@ 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: +```sh +python visualizer_drag_gradio.py --listen +``` +If you want to run DragGAN Gradio demo with given server port, for example port 8888: +```sh +python visualizer_drag_gradio.py --port=8888 +``` ## Acknowledgement diff --git a/visualizer_drag_gradio.py b/visualizer_drag_gradio.py index 42e6c15..1f27ab8 100644 --- a/visualizer_drag_gradio.py +++ b/visualizer_drag_gradio.py @@ -13,10 +13,15 @@ from gradio_utils import (ImageMask, draw_mask_on_image, draw_points_on_image, get_latest_points_pair, get_valid_mask, on_change_single_global_state) from viz.renderer import Renderer, add_watermark_np +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('--port', type=int, + help="launch gradio with given server port", default=None) args = parser.parse_args() cache_dir = args.cache_dir @@ -861,6 +866,11 @@ with gr.Blocks() as app: outputs=[global_state, form_image], ) +sig = inspect.signature(app.launch) +params = sig.parameters +def_server_name = params["server_name"].default +def_server_port = params["server_port"].default + gr.close_all() app.queue(concurrency_count=3, max_size=20) -app.launch(share=args.share) +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)