Commit
·
ab655eb
1
Parent(s):
dff0abc
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
from tts_infer.tts import TextToMel, MelToWav
|
| 2 |
from tts_infer.num_to_word_on_sent import normalize_nums
|
| 3 |
-
from ai4bharat.transliteration import XlitEngine
|
| 4 |
import gradio as gr
|
| 5 |
|
| 6 |
|
|
@@ -15,11 +15,11 @@ mel_to_wav_female = MelToWav(hifi_model_dir='checkpoints/hifi/female', device=de
|
|
| 15 |
text_to_mel_male = create_text_to_mel('checkpoints/glow/male')
|
| 16 |
mel_to_wav_male = MelToWav(hifi_model_dir='checkpoints/hifi/male', device=device)
|
| 17 |
|
| 18 |
-
def translit(text, lang):
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
|
| 24 |
|
| 25 |
|
|
@@ -27,7 +27,7 @@ def run_tts(text, selected_voice):
|
|
| 27 |
lang = "pa" # Punjabi language code
|
| 28 |
text = text.replace('।', '.')
|
| 29 |
text_num_to_word = normalize_nums(text, lang) # converting numbers to words in lang
|
| 30 |
-
text_num_to_word_and_transliterated = translit(text_num_to_word, lang) # transliterating English words to lang
|
| 31 |
|
| 32 |
if selected_voice == "Male Voice":
|
| 33 |
text_to_mel = text_to_mel_male
|
|
@@ -36,7 +36,7 @@ def run_tts(text, selected_voice):
|
|
| 36 |
text_to_mel = text_to_mel_female
|
| 37 |
mel_to_wav = mel_to_wav_female
|
| 38 |
|
| 39 |
-
mel = text_to_mel.generate_mel(
|
| 40 |
audio, sr = mel_to_wav.generate_wav(mel)
|
| 41 |
return sr, audio
|
| 42 |
|
|
|
|
| 1 |
from tts_infer.tts import TextToMel, MelToWav
|
| 2 |
from tts_infer.num_to_word_on_sent import normalize_nums
|
| 3 |
+
# from ai4bharat.transliteration import XlitEngine
|
| 4 |
import gradio as gr
|
| 5 |
|
| 6 |
|
|
|
|
| 15 |
text_to_mel_male = create_text_to_mel('checkpoints/glow/male')
|
| 16 |
mel_to_wav_male = MelToWav(hifi_model_dir='checkpoints/hifi/male', device=device)
|
| 17 |
|
| 18 |
+
# def translit(text, lang):
|
| 19 |
+
# engine = XlitEngine(lang)
|
| 20 |
+
# words = [engine.translit_word(word, topk=1)[lang][0] for word in text.split()]
|
| 21 |
+
# updated_sent = ' '.join(words)
|
| 22 |
+
# return updated_sent
|
| 23 |
|
| 24 |
|
| 25 |
|
|
|
|
| 27 |
lang = "pa" # Punjabi language code
|
| 28 |
text = text.replace('।', '.')
|
| 29 |
text_num_to_word = normalize_nums(text, lang) # converting numbers to words in lang
|
| 30 |
+
# text_num_to_word_and_transliterated = translit(text_num_to_word, lang) # transliterating English words to lang
|
| 31 |
|
| 32 |
if selected_voice == "Male Voice":
|
| 33 |
text_to_mel = text_to_mel_male
|
|
|
|
| 36 |
text_to_mel = text_to_mel_female
|
| 37 |
mel_to_wav = mel_to_wav_female
|
| 38 |
|
| 39 |
+
mel = text_to_mel.generate_mel(text_num_to_word)
|
| 40 |
audio, sr = mel_to_wav.generate_wav(mel)
|
| 41 |
return sr, audio
|
| 42 |
|