Support German TTS (#394)
This commit is contained in:
@@ -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
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -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) {
|
||||||
|
|||||||
@@ -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,
|
||||||
};
|
};
|
||||||
|
|||||||
Reference in New Issue
Block a user