Approximetal commited on
Commit
ad7caf7
Β·
verified Β·
1 Parent(s): 888f947

Update inference_gradio.py

Browse files
Files changed (1) hide show
  1. inference_gradio.py +14 -14
inference_gradio.py CHANGED
@@ -37,16 +37,16 @@ REPO_ROOT = Path(__file__).resolve().parent
37
 
38
  # HF location for pretrained assets (used as a fallback when local files are missing)
39
  HF_PRETRAINED_ROOT = "hf://LEMAS-Project/LEMAS-TTS/pretrained_models"
40
- CKPTS_ROOT = HF_PRETRAINED_ROOT / "ckpts"
41
 
42
  # 1) ζŒ‡ε‘δ½ δ»“εΊ“ι‡Œηš„ libespeak-ng.so
43
- ESPEAK_LIB = HF_PRETRAINED_ROOT / "espeak-ng-lib" / "libespeak-ng.so"
44
  os.environ["PHONEMIZER_ESPEAK_LIBRARY"] = str(ESPEAK_LIB)
45
 
46
  # 2) ζŒ‡ε‘δ½ δ»“εΊ“ι‡Œηš„ espeak-ng-data
47
- ESPEAK_DATA_DIR = HF_PRETRAINED_ROOT / "espeak-ng-data"
48
- os.environ["ESPEAK_DATA_PATH"] = str(ESPEAK_DATA_DIR)
49
- os.environ["ESPEAKNG_DATA_PATH"] = str(ESPEAK_DATA_DIR)
50
 
51
 
52
  class UVR5:
@@ -88,8 +88,8 @@ class UVR5:
88
  return output_audio.squeeze().T.numpy(), 44100
89
 
90
  denoise_model = UVR5(
91
- model_dir=HF_PRETRAINED_ROOT / "uvr5",
92
- code_dir=REPO_ROOT / "uvr5",
93
  )
94
 
95
  def load_wav(audio_info, sr=16000, channel=1):
@@ -479,13 +479,13 @@ with gr.Blocks(title="LEMAS-TTS Inference") as app:
479
 
480
  # Examples
481
  def _resolve_example(name: str) -> str:
482
- local = HF_PRETRAINED_ROOT / "data" / "test_examples" / name
483
- if local.is_file():
484
- return str(local)
485
  remote_map = {
486
- "en.wav": f"{HF_PRETRAINED_ROOT}/data/test_examples/en.wav",
487
- "es.wav": f"{HF_PRETRAINED_ROOT}/data/test_examples/es.wav",
488
- "pt.wav": f"{HF_PRETRAINED_ROOT}/data/test_examples/pt.wav",
489
  }
490
  url = remote_map.get(name)
491
  return str(cached_path(url)) if url is not None else ""
@@ -596,7 +596,7 @@ def main(port, host, share, api):
596
  server_port=port,
597
  share=share,
598
  show_api=api,
599
- allowed_paths=[str(HF_PRETRAINED_ROOT / "data")],
600
  )
601
 
602
 
 
37
 
38
  # HF location for pretrained assets (used as a fallback when local files are missing)
39
  HF_PRETRAINED_ROOT = "hf://LEMAS-Project/LEMAS-TTS/pretrained_models"
40
+ CKPTS_ROOT = os.path.join(HF_PRETRAINED_ROOT, "ckpts")
41
 
42
  # 1) ζŒ‡ε‘δ½ δ»“εΊ“ι‡Œηš„ libespeak-ng.so
43
+ ESPEAK_LIB = os.path.join(HF_PRETRAINED_ROOT, "espeak-ng-lib", "libespeak-ng.so")
44
  os.environ["PHONEMIZER_ESPEAK_LIBRARY"] = str(ESPEAK_LIB)
45
 
46
  # 2) ζŒ‡ε‘δ½ δ»“εΊ“ι‡Œηš„ espeak-ng-data
47
+ ESPEAK_DATA_DIR = os.path.join(HF_PRETRAINED_ROOT, "espeak-ng-data")
48
+ os.environ["ESPEAK_DATA_PATH"] = ESPEAK_DATA_DIR
49
+ os.environ["ESPEAKNG_DATA_PATH"] = ESPEAK_DATA_DIR
50
 
51
 
52
  class UVR5:
 
88
  return output_audio.squeeze().T.numpy(), 44100
89
 
90
  denoise_model = UVR5(
91
+ model_dir=os.path.join(HF_PRETRAINED_ROOT, "uvr5"),
92
+ code_dir=str(REPO_ROOT / "uvr5"),
93
  )
94
 
95
  def load_wav(audio_info, sr=16000, channel=1):
 
479
 
480
  # Examples
481
  def _resolve_example(name: str) -> str:
482
+ local = os.path.join(HF_PRETRAINED_ROOT, "data", "test_examples", name)
483
+ if os.path.isfile(local):
484
+ return local
485
  remote_map = {
486
+ "en.wav": os.path.join(HF_PRETRAINED_ROOT, "data", "test_examples", "en.wav"),
487
+ "es.wav": os.path.join(HF_PRETRAINED_ROOT, "data", "test_examples", "es.wav"),
488
+ "pt.wav": os.path.join(HF_PRETRAINED_ROOT, "data", "test_examples", "pt.wav"),
489
  }
490
  url = remote_map.get(name)
491
  return str(cached_path(url)) if url is not None else ""
 
596
  server_port=port,
597
  share=share,
598
  show_api=api,
599
+ allowed_paths=[str(os.path.join(HF_PRETRAINED_ROOT, "data"))],
600
  )
601
 
602