Update app.py

app.py

@@ -1,16 +1,32 @@
+# ===================================================================
+# [最重要] アプリケーションの本体を読み込む前に、モデルのセットアップを実行
+import initialize
+import time
+
+print("--- [1/3] Starting Model Setup Script ---")
+start_time = time.time()
+# Hugging Face Spacesでは推論のみ行うのが一般的なので、
+# only_infer=True を指定して不要なモデルのダウンロードをスキップします。
+# これにより起動が高速化します。
+initialize.main(only_infer=True) 
+end_time = time.time()
+print(f"--- [2/3] Model Setup Complete. Elapsed time: {end_time - start_time:.2f} seconds ---")
+# ===================================================================
+
+
+# [3/3] モデルの準備が完了したので、アプリケーションの本体を読み込んで起動
+print("--- [3/3] Initializing and launching Gradio App... ---")
+
 import argparse
 from pathlib import Path
 
 import gradio as gr
 import torch
 
-# --- [変更点1] 必要な関数だけをインポート ---
 # 拡張機能タブから必要なものだけをインポート
 from gradio_tabs.single import create_synthesis_app
 from gradio_tabs.merge import create_merge_app
 
-# ---------------------------------------------
-
 from config import get_path_config
 from style_bert_vits2.constants import GRADIO_THEME, VERSION
 from style_bert_vits2.nlp.japanese import pyopenjtalk_worker
@@ -27,37 +43,38 @@ update_dict()
 
 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=None)
+# Hugging Face Spacesで動かす場合、hostとportはGradioが自動で設定するため、
+# ここでの指定は通常不要です。
+# 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("--share", action="store_true", help="Set to True to create a public link")
 
 args = parser.parse_args()
 device = args.device
 if device == "cuda" and not torch.cuda.is_available():
+    print("CUDA not available, falling back to CPU.")
     device = "cpu"
+else:
+    print(f"Using device: {device}")
+
 
 path_config = get_path_config()
+
+# この時点で、initialize.pyが作成したモデルへのシンボリックリンクが
+# Path(path_config.assets_root) 以下に存在しているはずです。
 model_holder = TTSModelHolder(Path(path_config.assets_root), device)
 
 # 起動時にトップへスクロールさせるjsを追加
 with gr.Blocks(theme=GRADIO_THEME, js="() => { window.scrollTo(0, 0); }") as app:
-    # タイトルを少し変更しても良いかもしれません
     gr.Markdown(f"# Style-Bert-VITS2 拡張ツール (version {VERSION})")
     
-    # --- [変更点2] 表示したいタブだけを残す ---
     with gr.Tabs():
         with gr.Tab("拡張音声合成"):
-            # 拡張ツールの音声合成タブ (single.py)
             create_synthesis_app(model_holder=model_holder)
         with gr.Tab("モデルマージ"):
             create_merge_app(model_holder=model_holder)
 
-    # ---------------------------------------------
-
-app.launch(
-    server_name=args.host,
-    server_port=args.port,
-    inbrowser=not args.no_autolaunch,
-    share=args.share,
-)
+# Hugging Face Spacesでは、share=Trueは不要（自動で公開URLが生成される）
+# また、サーバー名とポートも指定しない方が安定します。
+app.launch()