Update inference_gradio.py

inference_gradio.py

@@ -14,7 +14,7 @@ from pathlib import Path
 
 from cached_path import cached_path
 
-from lemas_tts.api import TTS, PRETRAINED_ROOT, CKPTS_ROOT
+from lemas_tts.api import TTS
 
 # Global variables
 tts_api = None
@@ -37,13 +37,14 @@ REPO_ROOT = Path(__file__).resolve().parent
 
 # HF location for pretrained assets (used as a fallback when local files are missing)
 HF_PRETRAINED_ROOT = "hf://LEMAS-Project/LEMAS-TTS/pretrained_models"
+CKPTS_ROOT = HF_PRETRAINED_ROOT / "ckpts"
 
 # 1) 指向你仓库里的 libespeak-ng.so
-ESPEAK_LIB = PRETRAINED_ROOT / "espeak-ng-lib" / "libespeak-ng.so"
+ESPEAK_LIB = HF_PRETRAINED_ROOT / "espeak-ng-lib" / "libespeak-ng.so"
 os.environ["PHONEMIZER_ESPEAK_LIBRARY"] = str(ESPEAK_LIB)
 
 # 2) 指向你仓库里的 espeak-ng-data
-ESPEAK_DATA_DIR = PRETRAINED_ROOT / "espeak-ng-data"
+ESPEAK_DATA_DIR = HF_PRETRAINED_ROOT / "espeak-ng-data"
 os.environ["ESPEAK_DATA_PATH"] = str(ESPEAK_DATA_DIR)
 os.environ["ESPEAKNG_DATA_PATH"] = str(ESPEAK_DATA_DIR)
 
@@ -87,7 +88,7 @@ class UVR5:
         return output_audio.squeeze().T.numpy(), 44100
 
 denoise_model = UVR5(
-    model_dir=PRETRAINED_ROOT / "uvr5",
+    model_dir=HF_PRETRAINED_ROOT / "uvr5",
     code_dir=REPO_ROOT / "uvr5",
 )
 
@@ -177,7 +178,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(PRETRAINED_ROOT / "data"),
+        str(HF_PRETRAINED_ROOT / "data"),
     ]
     
     project_list = []
@@ -234,7 +235,7 @@ def infer(
         use_prosody_encoder = True if "prosody" in str(ckpt_resolved) else False
 
         # Resolve vocab file (local or HF)
-        local_vocab = Path(PRETRAINED_ROOT) / "data" / project / "vocab.txt"
+        local_vocab = Path(HF_PRETRAINED_ROOT) / "data" / project / "vocab.txt"
         if local_vocab.is_file():
             vocab_file = str(local_vocab)
         else:
@@ -478,7 +479,7 @@ with gr.Blocks(title="LEMAS-TTS Inference") as app:
 
     # Examples
     def _resolve_example(name: str) -> str:
-        local = PRETRAINED_ROOT / "data" / "test_examples" / name
+        local = HF_PRETRAINED_ROOT / "data" / "test_examples" / name
         if local.is_file():
             return str(local)
         remote_map = {
@@ -595,7 +596,7 @@ def main(port, host, share, api):
         server_port=port,
         share=share,
         show_api=api,
-        allowed_paths=[str(PRETRAINED_ROOT / "data")],
+        allowed_paths=[str(HF_PRETRAINED_ROOT / "data")],
     )