Feat: koharune-ami

app.py

@@ -11,6 +11,7 @@ from gradio_tabs.inference import create_inference_app
 from gradio_tabs.merge import create_merge_app
 from gradio_tabs.style_vectors import create_style_vectors_app
 from gradio_tabs.train import create_train_app
+from initialize import download_default_models
 from style_bert_vits2.constants import GRADIO_THEME, VERSION
 from style_bert_vits2.nlp.japanese import pyopenjtalk_worker
 from style_bert_vits2.nlp.japanese.user_dict import update_dict
@@ -30,12 +31,16 @@ parser.add_argument("--host", type=str, default="127.0.0.1")
 parser.add_argument("--port", type=int, default=None)
 parser.add_argument("--no_autolaunch", action="store_true")
 parser.add_argument("--share", action="store_true")
+parser.add_argument("--skip_default_models", action="store_true")
 
 args = parser.parse_args()
 device = args.device
 if device == "cuda" and not torch.cuda.is_available():
     device = "cpu"
 
+if not args.skip_default_models:
+    download_default_models()
+
 path_config = get_path_config()
 model_holder = TTSModelHolder(Path(path_config.assets_root), device)
 
@@ -55,7 +60,6 @@ with gr.Blocks(theme=GRADIO_THEME) as app:
         with gr.Tab("モデルダウンロード"):
             create_download_app()
 
-
 app.launch(
     server_name=args.host,
     server_port=args.port,