Skip to content

Commit

Permalink
Add Chinese character input support
Browse files Browse the repository at this point in the history
  • Loading branch information
oceanarium committed Aug 15, 2021
1 parent 96e9d74 commit aa1c345
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 12 deletions.
6 changes: 5 additions & 1 deletion synthesizer/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from typing import Union, List
import numpy as np
import librosa

from pypinyin import lazy_pinyin, Style

class Synthesizer:
sample_rate = hparams.sample_rate
Expand Down Expand Up @@ -91,6 +91,10 @@ def synthesize_spectrograms(self, texts: List[str],
simple_table([("Tacotron", str(tts_k) + "k"),
("r", self._model.r)])

#convert chinese char to pinyin
list_of_pinyin = lazy_pinyin(texts, style=Style.TONE3)
texts = [" ".join([v for v in list_of_pinyin if v.strip()])]

# Preprocess text inputs
inputs = [text_to_sequence(text.strip(), hparams.tts_cleaner_names) for text in texts]
if not isinstance(embeddings, list):
Expand Down
13 changes: 2 additions & 11 deletions toolbox/ui.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,17 +36,8 @@
], dtype=np.float) / 255

default_text = \
"Welcome to the toolbox! To begin, load an utterance from your datasets or record one " \
"yourself.\nOnce its embedding has been created, you can synthesize any text written here.\n" \
"The synthesizer expects to generate " \
"outputs that are somewhere between 5 and 12 seconds.\nTo mark breaks, write a new line. " \
"Each line will be treated separately.\nThen, they are joined together to make the final " \
"spectrogram. Use the vocoder to generate audio.\nThe vocoder generates almost in constant " \
"time, so it will be more time efficient for longer inputs like this one.\nOn the left you " \
"have the embedding projections. Load or record more utterances to see them.\nIf you have " \
"at least 2 or 3 utterances from a same speaker, a cluster should form.\nSynthesized " \
"utterances are of the same color as the speaker whose voice was used, but they're " \
"represented with a cross."
"欢迎使用工具箱, 现已支持中文输入!"



class UI(QDialog):
Expand Down

0 comments on commit aa1c345

Please sign in to comment.