Jobanpreet
commited on
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 |
|