Update app.py
Browse files
app.py
CHANGED
|
@@ -36,25 +36,7 @@ limitation = os.getenv("SYSTEM") == "spaces" # limit text and audio length in h
|
|
| 36 |
|
| 37 |
def create_tts_fn(model, hps, speaker_ids):
|
| 38 |
def tts_fn(text, speaker, language, speed, is_symbol):
|
| 39 |
-
|
| 40 |
-
text_len = len(re.sub("\[([A-Z]{2})\]", "", text))
|
| 41 |
-
max_len = 15000
|
| 42 |
-
if is_symbol:
|
| 43 |
-
max_len *= 3
|
| 44 |
-
if text_len > max_len:
|
| 45 |
-
return "Error: Text is too long", None
|
| 46 |
-
if language is not None:
|
| 47 |
-
text = language_marks[language] + text + language_marks[language]
|
| 48 |
-
speaker_id = speaker_ids[speaker]
|
| 49 |
-
stn_tst = get_text(text, hps, is_symbol)
|
| 50 |
-
with no_grad():
|
| 51 |
-
x_tst = stn_tst.unsqueeze(0)
|
| 52 |
-
x_tst_lengths = LongTensor([stn_tst.size(0)])
|
| 53 |
-
sid = LongTensor([speaker_id])
|
| 54 |
-
audio = model.infer(x_tst, x_tst_lengths, sid=sid, noise_scale=.667, noise_scale_w=0.8,
|
| 55 |
-
length_scale=1.0 / speed)[0][0, 0].data.cpu().float().numpy()
|
| 56 |
-
del stn_tst, x_tst, x_tst_lengths, sid
|
| 57 |
-
return "Success", (hps.data.sampling_rate, audio)
|
| 58 |
|
| 59 |
return tts_fn
|
| 60 |
|
|
|
|
| 36 |
|
| 37 |
def create_tts_fn(model, hps, speaker_ids):
|
| 38 |
def tts_fn(text, speaker, language, speed, is_symbol):
|
| 39 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
|
| 41 |
return tts_fn
|
| 42 |
|