Support configuring gradio startup parameters

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
 

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)