From 07f766c19fe5346af7ffc5be221c93335aa378d3 Mon Sep 17 00:00:00 2001 From: Mert Cobanov Date: Thu, 29 Jun 2023 13:54:27 +0300 Subject: [PATCH] Open a port from docker container to local --- README.md | 4 ++-- visualizer_drag_gradio.py | 7 ++++++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 8fdd6dc..7b21594 100644 --- a/README.md +++ b/README.md @@ -75,8 +75,8 @@ Provided docker image is based on NGC PyTorch repository. To quickly try out vis ```sh docker build . -t draggan:latest -docker run -v "$PWD":/workspace/src -it draggan:latest bash -cd src && python visualizer_drag_gradio.py +docker run -p 7860: 7860 -v "$PWD":/workspace/src -it draggan:latest bash +cd src && python visualizer_drag_gradio.py --listen ``` Now you can open a shared link from Gradio (printed in the terminal console). Beware the Docker image takes about 25GB of disk space! diff --git a/visualizer_drag_gradio.py b/visualizer_drag_gradio.py index 42e6c15..ff9abe7 100644 --- a/visualizer_drag_gradio.py +++ b/visualizer_drag_gradio.py @@ -17,6 +17,11 @@ 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", +) args = parser.parse_args() cache_dir = args.cache_dir @@ -863,4 +868,4 @@ def on_click_show_mask(global_state, show_mask): 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 "127.0.0.1")