Spaces:
Running
on
Zero
Running
on
Zero
Update inference_gradio.py
Browse files- 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
|
| 41 |
|
| 42 |
# 1) ζεδ½ δ»εΊιη libespeak-ng.so
|
| 43 |
-
ESPEAK_LIB = HF_PRETRAINED_ROOT
|
| 44 |
os.environ["PHONEMIZER_ESPEAK_LIBRARY"] = str(ESPEAK_LIB)
|
| 45 |
|
| 46 |
# 2) ζεδ½ δ»εΊιη espeak-ng-data
|
| 47 |
-
ESPEAK_DATA_DIR = HF_PRETRAINED_ROOT
|
| 48 |
-
os.environ["ESPEAK_DATA_PATH"] =
|
| 49 |
-
os.environ["ESPEAKNG_DATA_PATH"] =
|
| 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
|
| 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
|
| 483 |
-
if
|
| 484 |
-
return
|
| 485 |
remote_map = {
|
| 486 |
-
"en.wav":
|
| 487 |
-
"es.wav":
|
| 488 |
-
"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
|
| 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 |
|