Update inference_gradio.py

inference_gradio.py

@@ -34,17 +34,20 @@ device = (
 )
 
 REPO_ROOT = Path(__file__).resolve().parent
+# Local pretrained root (used when running from a repo / Space that bundles weights)
+PRETRAINED_ROOT = str(REPO_ROOT / "pretrained_models")
+CKPTS_ROOT = os.path.join(PRETRAINED_ROOT, "ckpts")
 
 # HF location for pretrained assets (used as a fallback when local files are missing)
-HF_PRETRAINED_ROOT = "hf://LEMAS-Project/LEMAS-TTS/blob/main/pretrained_models"
-CKPTS_ROOT = os.path.join(HF_PRETRAINED_ROOT, "ckpts")
+# HF_PRETRAINED_ROOT = "hf://LEMAS-Project/LEMAS-TTS/blob/main/pretrained_models"
+
 
 # 1) 指向你仓库里的 libespeak-ng.so
-ESPEAK_LIB = os.path.join(HF_PRETRAINED_ROOT, "espeak-ng-lib", "libespeak-ng.so")
+ESPEAK_LIB = os.path.join(PRETRAINED_ROOT, "espeak-ng-lib", "libespeak-ng.so")
 os.environ["PHONEMIZER_ESPEAK_LIBRARY"] = str(ESPEAK_LIB)
 
 # 2) 指向你仓库里的 espeak-ng-data
-ESPEAK_DATA_DIR = os.path.join(HF_PRETRAINED_ROOT, "espeak-ng-data")
+ESPEAK_DATA_DIR = os.path.join(PRETRAINED_ROOT, "espeak-ng-data")
 os.environ["ESPEAK_DATA_PATH"] = ESPEAK_DATA_DIR
 os.environ["ESPEAKNG_DATA_PATH"] = ESPEAK_DATA_DIR
 
@@ -88,7 +91,7 @@ class UVR5:
         return output_audio.squeeze().T.numpy(), 44100
 
 denoise_model = UVR5(
-    model_dir=os.path.join(HF_PRETRAINED_ROOT, "uvr5"),
+    model_dir=os.path.join(PRETRAINED_ROOT, "uvr5"),
     code_dir=str(REPO_ROOT / "uvr5"),
 )
 
@@ -124,8 +127,8 @@ def get_checkpoints_project(project_name=None, is_gradio=True):
     checkpoint_dir = [str(CKPTS_ROOT)]
     # Remote ckpt locations on HF (used if local ckpts are not present)
     remote_ckpts = {
-        "multilingual_grl": f"{HF_PRETRAINED_ROOT}/ckpts/multilingual_grl/multilingual_grl.safetensors",
-        "multilingual_prosody": f"{HF_PRETRAINED_ROOT}/ckpts/multilingual_prosody/multilingual_prosody.safetensors",
+        "multilingual_grl": f"{PRETRAINED_ROOT}/ckpts/multilingual_grl/multilingual_grl.safetensors",
+        "multilingual_prosody": f"{PRETRAINED_ROOT}/ckpts/multilingual_prosody/multilingual_prosody.safetensors",
     }
 
     if project_name is None:
@@ -178,7 +181,7 @@ def get_checkpoints_project(project_name=None, is_gradio=True):
 def get_available_projects():
     """Get available project names from data directory"""
     data_paths = [
-        str(HF_PRETRAINED_ROOT / "data"),
+        str(PRETRAINED_ROOT / "data"),
     ]
     
     project_list = []
@@ -235,13 +238,13 @@ def infer(
         use_prosody_encoder = True if "prosody" in str(ckpt_resolved) else False
 
         # Resolve vocab file (local or HF)
-        local_vocab = Path(HF_PRETRAINED_ROOT) / "data" / project / "vocab.txt"
+        local_vocab = Path(PRETRAINED_ROOT) / "data" / project / "vocab.txt"
         if local_vocab.is_file():
             vocab_file = str(local_vocab)
         else:
             remote_vocab_map = {
-                "multilingual_grl": f"{HF_PRETRAINED_ROOT}/data/multilingual_grl/vocab.txt",
-                "multilingual_prosody": f"{HF_PRETRAINED_ROOT}/data/multilingual_prosody/vocab.txt",
+                "multilingual_grl": f"{PRETRAINED_ROOT}/data/multilingual_grl/vocab.txt",
+                "multilingual_prosody": f"{PRETRAINED_ROOT}/data/multilingual_prosody/vocab.txt",
             }
             remote_vocab = remote_vocab_map.get(project)
             if remote_vocab is None:
@@ -259,13 +262,13 @@ def infer(
             prosody_cfg_path = str(local_prosody_cfg)
         else:
             prosody_cfg_path = str(
-                cached_path(f"{HF_PRETRAINED_ROOT}/ckpts/prosody_encoder/pretssel_cfg.json")
+                cached_path(f"{PRETRAINED_ROOT}/ckpts/prosody_encoder/pretssel_cfg.json")
             )
         if local_prosody_ckpt.is_file():
             prosody_ckpt_path = str(local_prosody_ckpt)
         else:
             prosody_ckpt_path = str(
-                cached_path(f"{HF_PRETRAINED_ROOT}/ckpts/prosody_encoder/prosody_encoder_UnitY2.pt")
+                cached_path(f"{PRETRAINED_ROOT}/ckpts/prosody_encoder/prosody_encoder_UnitY2.pt")
             )
 
         try:
@@ -479,13 +482,13 @@ with gr.Blocks(title="LEMAS-TTS Inference") as app:
 
     # Examples
     def _resolve_example(name: str) -> str:
-        local = os.path.join(HF_PRETRAINED_ROOT, "data", "test_examples", name)
+        local = os.path.join(PRETRAINED_ROOT, "data", "test_examples", name)
         if os.path.isfile(local):
             return local
         remote_map = {
-            "en.wav": os.path.join(HF_PRETRAINED_ROOT, "data", "test_examples", "en.wav"),
-            "es.wav": os.path.join(HF_PRETRAINED_ROOT, "data", "test_examples", "es.wav"),
-            "pt.wav": os.path.join(HF_PRETRAINED_ROOT, "data", "test_examples", "pt.wav"),
+            "en.wav": os.path.join(PRETRAINED_ROOT, "data", "test_examples", "en.wav"),
+            "es.wav": os.path.join(PRETRAINED_ROOT, "data", "test_examples", "es.wav"),
+            "pt.wav": os.path.join(PRETRAINED_ROOT, "data", "test_examples", "pt.wav"),
         }
         url = remote_map.get(name)
         return str(cached_path(url)) if url is not None else ""
@@ -596,7 +599,7 @@ def main(port, host, share, api):
         server_port=port,
         share=share,
         show_api=api,
-        allowed_paths=[str(os.path.join(HF_PRETRAINED_ROOT, "data"))],
+        allowed_paths=[str(os.path.join(PRETRAINED_ROOT, "data"))],
     )