diff --git a/optispeech/text/tokenizers.py b/optispeech/text/tokenizers.py index 3466206..8c93ed7 100644 --- a/optispeech/text/tokenizers.py +++ b/optispeech/text/tokenizers.py @@ -1,4 +1,5 @@ from abc import ABC, abstractmethod +from os import getenv from . import symbols from .normalization import UNICODE_NORM_FORM, collapse_whitespace, intersperse, preprocess_text @@ -93,5 +94,6 @@ def phonemize_text(self, text: str, language: str) -> str: # Preprocess text = self.preprocess_text(text, language) # Phonemize - phonemes = phonemize_espeak(text, language) + espeak_data_path = getenv('ESPEAK_DATA_PATH') + phonemes = phonemize_espeak(text, language, data_path=espeak_data_path) return phonemes, text diff --git a/ospeech/ospeech/text/tokenizers.py b/ospeech/ospeech/text/tokenizers.py index 3466206..8c93ed7 100644 --- a/ospeech/ospeech/text/tokenizers.py +++ b/ospeech/ospeech/text/tokenizers.py @@ -1,4 +1,5 @@ from abc import ABC, abstractmethod +from os import getenv from . import symbols from .normalization import UNICODE_NORM_FORM, collapse_whitespace, intersperse, preprocess_text @@ -93,5 +94,6 @@ def phonemize_text(self, text: str, language: str) -> str: # Preprocess text = self.preprocess_text(text, language) # Phonemize - phonemes = phonemize_espeak(text, language) + espeak_data_path = getenv('ESPEAK_DATA_PATH') + phonemes = phonemize_espeak(text, language, data_path=espeak_data_path) return phonemes, text