tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Add: --host/--port option to app.py to allow specifying listening host/port

Browse Source
This commit is contained in:
tsukumi
2024-03-08 22:14:39 +00:00
parent d22a11ebb2
commit b98fecb46d
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app.py
View File

@@ -24,6 +24,8 @@ with Path("configs/paths.yml").open("r", encoding="utf-8") as f:
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--device", type=str, default="cuda") 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("--no_autolaunch", action="store_true")
parser.add_argument("--share", 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) 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)

1
webui/train.py
View File

@@ -792,5 +792,4 @@ def create_train_app():
outputs=[use_jp_extra_train], outputs=[use_jp_extra_train],
) )
# app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name)
return app return app