Support German TTS (#394)

This commit is contained in:
Fangjun Kuang
2023-10-27 11:12:45 +08:00
committed by GitHub
parent 93ef4ee4bc
commit fbf4c903e1
3 changed files with 15 additions and 12 deletions

View File

@@ -77,20 +77,13 @@ class MainActivity : AppCompatActivity() {
return return
} }
Toast.makeText(applicationContext, "Generating...Please wait", Toast.LENGTH_LONG).show() play.isEnabled = false;
val audio = tts.generate(text = textStr, sid = sidInt, speed = speedFloat) val audio = tts.generate(text = textStr, sid = sidInt, speed = speedFloat)
val filename = application.filesDir.absolutePath + "/generated.wav" val filename = application.filesDir.absolutePath + "/generated.wav"
val ok = audio.samples.size > 0 && audio.save(filename) val ok = audio.samples.size > 0 && audio.save(filename)
if (ok) { if (ok) {
play.isEnabled = true play.isEnabled = true
Toast.makeText(
applicationContext,
"Generated! Please click play to listen to it",
Toast.LENGTH_LONG
).show()
} else {
play.isEnabled = false
} }
} }

View File

@@ -129,10 +129,12 @@ std::vector<int64_t> Lexicon::ConvertTextToTokenIds(
switch (language_) { switch (language_) {
case Language::kEnglish: case Language::kEnglish:
return ConvertTextToTokenIdsEnglish(text); return ConvertTextToTokenIdsEnglish(text);
case Language::kGerman:
return ConvertTextToTokenIdsGerman(text);
case Language::kChinese: case Language::kChinese:
return ConvertTextToTokenIdsChinese(text); return ConvertTextToTokenIdsChinese(text);
default: default:
SHERPA_ONNX_LOGE("Unknonw language: %d", static_cast<int32_t>(language_)); SHERPA_ONNX_LOGE("Unknown language: %d", static_cast<int32_t>(language_));
exit(-1); exit(-1);
} }
@@ -246,6 +248,8 @@ void Lexicon::InitLanguage(const std::string &_lang) {
ToLowerCase(&lang); ToLowerCase(&lang);
if (lang == "english") { if (lang == "english") {
language_ = Language::kEnglish; language_ = Language::kEnglish;
} else if (lang == "german") {
language_ = Language::kGerman;
} else if (lang == "chinese") { } else if (lang == "chinese") {
language_ = Language::kChinese; language_ = Language::kChinese;
} else { } else {
@@ -269,8 +273,8 @@ void Lexicon::InitLexicon(std::istream &is) {
ToLowerCase(&word); ToLowerCase(&word);
if (word2ids_.count(word)) { if (word2ids_.count(word)) {
SHERPA_ONNX_LOGE("Duplicated word: %s", word.c_str()); SHERPA_ONNX_LOGE("Duplicated word: %s. Ignore it.", word.c_str());
return; continue;
} }
while (iss >> phone) { while (iss >> phone) {

View File

@@ -36,6 +36,11 @@ class Lexicon {
std::vector<int64_t> ConvertTextToTokenIds(const std::string &text) const; std::vector<int64_t> ConvertTextToTokenIds(const std::string &text) const;
private: private:
std::vector<int64_t> ConvertTextToTokenIdsGerman(
const std::string &text) const {
return ConvertTextToTokenIdsEnglish(text);
}
std::vector<int64_t> ConvertTextToTokenIdsEnglish( std::vector<int64_t> ConvertTextToTokenIdsEnglish(
const std::string &text) const; const std::string &text) const;
@@ -50,6 +55,7 @@ class Lexicon {
private: private:
enum class Language { enum class Language {
kEnglish, kEnglish,
kGerman,
kChinese, kChinese,
kUnknown, kUnknown,
}; };