harveen commited on
Commit
e22a898
1 Parent(s): 3704e25
Files changed (2) hide show
  1. app.py +7 -4
  2. ttsv/utils/inference/run_gradio.py +1 -1
app.py CHANGED
@@ -1,6 +1,9 @@
1
  import os
2
- os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/hindi/hi-IN/female_voice_0/glow.zip && unzip -q glow.zip -d ttsv/checkpoints/')
3
- os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/hindi/hi-IN/female_voice_0/hifi.zip && unzip -q hifi.zip -d ttsv/checkpoints/')
 
 
 
4
  os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/translit_models.zip -P ttsv/checkpoints/ && unzip -q ttsv/checkpoints/translit_models.zip -d ttsv/checkpoints/')
5
 
6
 
@@ -16,8 +19,8 @@ from argparse import Namespace
16
 
17
 
18
  args = {
19
- 'acoustic':'/home/user/app/ttsv/checkpoints/glow_ckp',
20
- 'vocoder':'/home/user/app/ttsv/checkpoints/hifi_ckp',
21
  'device':'cpu',
22
  'lang':'hi'
23
  }
 
1
  import os
2
+ os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/hindi/hi-IN/female_voice_0/glow.zip && unzip -q glow.zip -d ttsv/checkpoints/female')
3
+ os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/hindi/hi-IN/female_voice_0/hifi.zip && unzip -q hifi.zip -d ttsv/checkpoints/female')
4
+ os.system('rm glow.zip && rm hifi.zip')
5
+ os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/hindi/hi-IN/male_voice_1/glow.zip && unzip -q glow.zip -d ttsv/checkpoints/male')
6
+ os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/hindi/hi-IN/male_voice_1/hifi.zip && unzip -q hifi.zip -d ttsv/checkpoints/male')
7
  os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/translit_models.zip -P ttsv/checkpoints/ && unzip -q ttsv/checkpoints/translit_models.zip -d ttsv/checkpoints/')
8
 
9
 
 
19
 
20
 
21
  args = {
22
+ 'acoustic':'/home/user/app/ttsv/checkpoints/female/glow_ckp,/home/user/app/ttsv/checkpoints/male/glow_ckp',
23
+ 'vocoder':'/home/user/app/ttsv/checkpoints/female/hifi_ckp,/home/user/app/ttsv/checkpoints/male/hifi_ckp',
24
  'device':'cpu',
25
  'lang':'hi'
26
  }
ttsv/utils/inference/run_gradio.py CHANGED
@@ -43,7 +43,7 @@ def build_gradio(args):
43
 
44
  inputs_to_gradio = [textbox, gender, slider_noise_scale, slider_length_sclae, choice_transliteration, choice_number_conversion, choice_split_sentences]
45
  iface = gr.Interface(fn=hit_tts, inputs=inputs_to_gradio, outputs=op, theme='huggingface', title='Vakyansh Hindi TTS', article = 'Note: Transliteration models may not work well in some scenarios which can hamper the TTS quality, to evaluate the model in better sense it is advisable to provide input in the required langauge and switch off transliteration.')
46
- iface.launch(share=True, enable_queue=True)
47
 
48
  if __name__ == "__main__":
49
  parser = argparse.ArgumentParser()
 
43
 
44
  inputs_to_gradio = [textbox, gender, slider_noise_scale, slider_length_sclae, choice_transliteration, choice_number_conversion, choice_split_sentences]
45
  iface = gr.Interface(fn=hit_tts, inputs=inputs_to_gradio, outputs=op, theme='huggingface', title='Vakyansh Hindi TTS', article = 'Note: Transliteration models may not work well in some scenarios which can hamper the TTS quality, to evaluate the model in better sense it is advisable to provide input in the required langauge and switch off transliteration.')
46
+ iface.launch(enable_queue=True)
47
 
48
  if __name__ == "__main__":
49
  parser = argparse.ArgumentParser()