akadmin/openwakeword
1
0
Fork 0
mirror of https://github.com/dscripka/openWakeWord.git synced 2026-08-27 18:17:20 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixed bug when downloading phonemizer model [skip ci]

Browse source
This commit is contained in:
dscripka 2023-10-12 12:01:30 -04:00
parent a99d5f732f
commit 9a23c43ab4
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
openwakeword/data.py
View file

@ -925,6 +925,8 @@ def generate_adversarial_texts(input_text: str, N: int, include_partial_phrase:
# Download phonemizer model for OOV words, if needed
if [] in input_text_phones:
phonemizer_mdl_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "resources", "en_us_cmudict_forward.pt")
if not os.path.exists(os.path.join(os.path.dirname(os.path.abspath(__file__)), "resources")):
os.mkdir(os.path.join(os.path.dirname(os.path.abspath(__file__)), "resources"))
if not os.path.exists(phonemizer_mdl_path):
logging.warning("Downloading phonemizer model from DeepPhonemizer library...")
import requests