storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-21 08:06:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix apostrophe, hyphen

Browse Source
This commit is contained in:
Brandon Thomas
2022-07-05 03:31:25 -04:00
parent c81f8d323c
commit ffbf36fd13
4 changed files with 35 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/text/pipeline_components/phonetization/base_phonetization.py
+5 -1
View File
@@ -4,7 +4,11 @@ from typing import List
from text.pipeline_components.phonetization.cmudict import CMUDict
# Regex taken from open source tacotron2 extensions
_arpabet_split_re = re.compile(r'[^\s]*{.*?}[^\s]*|[\w]+|[^\w\s]+')
# NB: This was broken and couldn't handle apostrophes
#_arpabet_split_re = re.compile(r'[^\s]*{.*?}[^\s]*|[\w]+|[^\w\s]+')
# Modified regex to not split apostrophe (eg. "you're", "I'm", etc.) or hyphenation (eg "avant-garde")
_arpabet_split_re = re.compile(r'[^\s]*{.*?}[^\s]*|[\w]+\'[\w]+|[\w]+\-[\w]+|[\w]+|[^\w\s]+')
class BasePhonetization:
def __init__(self, phoneme_dictionary: CMUDict):