Jobanpreet commited on
Commit
ab655eb
1 Parent(s): dff0abc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
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
- 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,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(text_num_to_word_and_transliterated)
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