diff --git a/app.py b/app.py index c1250b0..81f0cb9 100644 --- a/app.py +++ b/app.py @@ -244,6 +244,18 @@ if __name__ == "__main__": parser.add_argument( "--share", action="store_true", help="Share this app publicly", default=False ) + parser.add_argument( + "--server-name", + type=str, + default=None, + help="Server name for Gradio app", + ) + parser.add_argument( + "--no-autolaunch", + action="store_true", + default=False, + help="Do not launch app automatically", + ) args = parser.parse_args() model_dir = args.dir @@ -435,4 +447,6 @@ if __name__ == "__main__": outputs=[style, ref_audio_path], ) - app.launch(inbrowser=True, share=args.share) + app.launch( + inbrowser=not args.no_autolaunch, share=args.share, server_name=args.server_name + ) diff --git a/webui_dataset.py b/webui_dataset.py index ffca25f..ca50d1b 100644 --- a/webui_dataset.py +++ b/webui_dataset.py @@ -1,3 +1,4 @@ +import argparse import os import gradio as gr @@ -192,4 +193,19 @@ with gr.Blocks(theme="NoCrypt/miku") as app: outputs=[result2], ) -app.launch(inbrowser=True) +parser = argparse.ArgumentParser() +parser.add_argument( + "--server-name", + type=str, + default=None, + help="Server name for Gradio app", +) +parser.add_argument( + "--no-autolaunch", + action="store_true", + default=False, + help="Do not launch app automatically", +) +args = parser.parse_args() + +app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name) diff --git a/webui_merge.py b/webui_merge.py index 2635c5a..5a36442 100644 --- a/webui_merge.py +++ b/webui_merge.py @@ -1,3 +1,4 @@ +import argparse import json import os import sys @@ -395,5 +396,19 @@ with gr.Blocks(theme="NoCrypt/miku") as app: outputs=[audio_output], ) +parser = argparse.ArgumentParser() +parser.add_argument( + "--server-name", + type=str, + default=None, + help="Server name for Gradio app", +) +parser.add_argument( + "--no-autolaunch", + action="store_true", + default=False, + help="Do not launch app automatically", +) +args = parser.parse_args() -app.launch(inbrowser=True) +app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name) diff --git a/webui_style_vectors.py b/webui_style_vectors.py index 5630b52..951f845 100644 --- a/webui_style_vectors.py +++ b/webui_style_vectors.py @@ -1,3 +1,4 @@ +import argparse import json import os import shutil @@ -443,4 +444,20 @@ with gr.Blocks(theme="NoCrypt/miku") as app: inputs=[model_name, audio_files_text, style_names_text], outputs=[info2], ) - app.launch(inbrowser=True) + +parser = argparse.ArgumentParser() +parser.add_argument( + "--server-name", + type=str, + default=None, + help="Server name for Gradio app", +) +parser.add_argument( + "--no-autolaunch", + action="store_true", + default=False, + help="Do not launch app automatically", +) +args = parser.parse_args() + +app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name) diff --git a/webui_train.py b/webui_train.py index f2a582f..8694215 100644 --- a/webui_train.py +++ b/webui_train.py @@ -1,6 +1,8 @@ +import argparse import json import os import shutil +from datetime import datetime from multiprocessing import cpu_count import gradio as gr @@ -8,8 +10,6 @@ import yaml from common.log import logger from common.subprocess_utils import run_script_with_log, second_elem_of -from datetime import datetime - logger_handler = None @@ -496,4 +496,19 @@ if __name__ == "__main__": second_elem_of(train), inputs=[model_name], outputs=[info_train] ) - app.launch(inbrowser=True) + parser = argparse.ArgumentParser() + parser.add_argument( + "--server-name", + type=str, + default=None, + help="Server name for Gradio app", + ) + parser.add_argument( + "--no-autolaunch", + action="store_true", + default=False, + help="Do not launch app automatically", + ) + args = parser.parse_args() + + app.launch(inbrowser=args.no_autolaunch, server_name=args.server_name)