From b98fecb46d18108ea1c5fdd36a256dcf7e8f655d Mon Sep 17 00:00:00 2001 From: tsukumi Date: Fri, 8 Mar 2024 22:14:39 +0000 Subject: [PATCH] Add: --host/--port option to app.py to allow specifying listening host/port --- app.py | 4 +++- webui/train.py | 1 - 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/app.py b/app.py index c2237b5..f0f94d0 100644 --- a/app.py +++ b/app.py @@ -24,6 +24,8 @@ with Path("configs/paths.yml").open("r", encoding="utf-8") as f: parser = argparse.ArgumentParser() parser.add_argument("--device", type=str, default="cuda") +parser.add_argument("--host", type=str, default="127.0.0.1") +parser.add_argument("--port", type=int, default=7860) parser.add_argument("--no_autolaunch", action="store_true") parser.add_argument("--share", action="store_true") @@ -49,4 +51,4 @@ with gr.Blocks(theme=GRADIO_THEME) as app: create_merge_app(model_holder=model_holder) -app.launch(inbrowser=not args.no_autolaunch, share=args.share) +app.launch(server_name=args.host, server_port=args.port, inbrowser=not args.no_autolaunch, share=args.share) diff --git a/webui/train.py b/webui/train.py index fa29719..d837471 100644 --- a/webui/train.py +++ b/webui/train.py @@ -792,5 +792,4 @@ def create_train_app(): outputs=[use_jp_extra_train], ) - # app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name) return app