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

refactor

Browse Source
This commit is contained in:
Brandon Thomas
2022-07-04 02:55:52 -04:00
parent 51544ea062
commit 951a34894b
5 changed files with 24 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/text/pipeline/english_pipeline_v1.py
+2 -4
View File
@@ -1,13 +1,11 @@
from typing import List
from tts.text.pipeline.base_pipeline import BasePipeline
from tts.text.pipeline_components.input_parsing.parser import ArpabetSequence, GraphemeSequence, ParsedSequence, parse_grapheme_and_arpabet_sequence
from tts.text.pipeline_components.normalization.english_normalization import EnglishNormalization
from tts.text.pipeline_components.normalization.english_normalization import normalize_english
from tts.text.pipeline_components.phonetization.cmudict import DEFAULT_CMUDICT_PATH, CMUDict
from tts.text.pipeline_components.phonetization.english_phonetization import EnglishPhonetization
from tts.text.pipeline_components.sequencing.english_symbols_v1 import SPACE_INTEGER_ENCODING, arpabet_to_sequence, symbols_to_sequence
_ENGLISH_NORMALIZER = EnglishNormalization()
_ENGLISH_DICTIONARY = CMUDict(DEFAULT_CMUDICT_PATH)
_ENGLISH_PHONETIZER = EnglishPhonetization(_ENGLISH_DICTIONARY)
@@ -27,7 +25,7 @@ class EnglishPipelineV1(BasePipeline):
for i in range(len(parsed_sequence)):
subsequence = parsed_sequence[i]
if isinstance(subsequence, GraphemeSequence):
normalized = _ENGLISH_NORMALIZER.normalize(subsequence.sequence)
normalized = normalize_english(subsequence.sequence)
parsed_sequence[i] = GraphemeSequence(normalized)