Commit
·
386d2a7
1
Parent(s):
42c9a4a
add japanese support
Browse files
app.py
CHANGED
|
@@ -53,6 +53,12 @@ model_path = os.path.join(
|
|
| 53 |
)
|
| 54 |
config = XttsConfig()
|
| 55 |
config.load_json(os.path.join(model_path, "config.json"))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
model = Xtts.init_from_config(config)
|
| 57 |
model.load_checkpoint(
|
| 58 |
config,
|
|
@@ -178,7 +184,7 @@ def wave_header_chunk(frame_input=b"", channels=1, sample_width=2, sample_rate=2
|
|
| 178 |
wav_buf.seek(0)
|
| 179 |
return wav_buf.read()
|
| 180 |
|
| 181 |
-
xtts_supported_languages=["en","es","fr","de","it","pt","pl","tr","ru","nl","cs","ar","zh-cn"]
|
| 182 |
def detect_language(prompt):
|
| 183 |
# Fast language autodetection
|
| 184 |
if len(prompt)>15:
|
|
@@ -186,6 +192,10 @@ def detect_language(prompt):
|
|
| 186 |
if language_predicted == "zh":
|
| 187 |
#we use zh-cn on xtts
|
| 188 |
language_predicted = "zh-cn"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 189 |
if language_predicted not in xtts_supported_languages:
|
| 190 |
print(f"Detected a language not supported by xtts :{language_predicted}, switching to english for now")
|
| 191 |
gr.Warning(f"Language detected '{language_predicted}' can not be spoken properly 'yet' ")
|
|
|
|
| 53 |
)
|
| 54 |
config = XttsConfig()
|
| 55 |
config.load_json(os.path.join(model_path, "config.json"))
|
| 56 |
+
|
| 57 |
+
if "ja-jp" not in config.languages:
|
| 58 |
+
#fix to have JP before next TTS update
|
| 59 |
+
config.languages.append("ja-jp")
|
| 60 |
+
|
| 61 |
+
|
| 62 |
model = Xtts.init_from_config(config)
|
| 63 |
model.load_checkpoint(
|
| 64 |
config,
|
|
|
|
| 184 |
wav_buf.seek(0)
|
| 185 |
return wav_buf.read()
|
| 186 |
|
| 187 |
+
xtts_supported_languages=["en","es","fr","de","it","pt","pl","tr","ru","nl","cs","ar","zh-cn","ja-jp"]
|
| 188 |
def detect_language(prompt):
|
| 189 |
# Fast language autodetection
|
| 190 |
if len(prompt)>15:
|
|
|
|
| 192 |
if language_predicted == "zh":
|
| 193 |
#we use zh-cn on xtts
|
| 194 |
language_predicted = "zh-cn"
|
| 195 |
+
if language_predicted == "ja":
|
| 196 |
+
#we use zh-cn
|
| 197 |
+
language_predicted = "ja-jp"
|
| 198 |
+
|
| 199 |
if language_predicted not in xtts_supported_languages:
|
| 200 |
print(f"Detected a language not supported by xtts :{language_predicted}, switching to english for now")
|
| 201 |
gr.Warning(f"Language detected '{language_predicted}' can not be spoken properly 'yet' ")
|