S1m0neAI commited on
Commit
c33802f
1 Parent(s): d4bd320

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -1,12 +1,13 @@
1
  import gradio as gr
2
  from infer_onnx import TTS
3
- from ruaccent import RUAccent # https://huggingface.co/TeraTTS/accentuator
4
 
5
  # Заголовок и ссылка на репозиторий с моделями
6
- title = "GitHub with models: https://github.com/Tera2Space/TeraTTS"
 
7
 
8
  # Список моделей TTS для выбора
9
- models = ["TeraTTS/natasha-g2p-vits", "TeraTTS/glados2-g2p-vits", "TeraTTS/glados-g2p-vits", "TeraTTS/girl_nice-g2p-vits"]
10
 
11
  # Создаем словарь моделей и инициализируем их
12
  models = {k: TTS(k) for k in models}
@@ -27,13 +28,13 @@ def text_to_speech(model_name, length_scale, text, prep_text):
27
  return 'temp.wav', f"Обработанный текст: '{text}'"
28
 
29
  # Создание интерфейса Gradio
30
- model_choice = gr.Dropdown(choices=list(models.keys()), value="TeraTTS/natasha-g2p-vits", label="Выберите модель")
31
  input_text = gr.Textbox(label="Введите текст для синтеза речи")
32
  prep_text = gr.Checkbox(label="Предобработать", info="Хотите предобработать текст? (ударения, ё)", value=True)
33
  length_scale = gr.Slider(minimum=0.1, maximum=2.0, label="Length scale (увеличить длину звучания) По умолчанию: 1.2", value=1.2)
34
 
35
  output_audio = gr.Audio(label="Аудио", type="numpy")
36
- output_text = gr.Textbox(label="Обработанный текст")
37
 
38
  iface = gr.Interface(fn=text_to_speech, inputs=[model_choice, length_scale, input_text, prep_text], outputs=[output_audio, output_text], title=title)
39
  iface.launch()
 
1
  import gradio as gr
2
  from infer_onnx import TTS
3
+ from ruaccent import RUAccent
4
 
5
  # Заголовок и ссылка на репозиторий с моделями
6
+ title = "RU - Russian Voices"
7
+ article="Speech synthesis - RU"
8
 
9
  # Список моделей TTS для выбора
10
+ models = ["TeraTTS/natasha-g2p-vits"]
11
 
12
  # Создаем словарь моделей и инициализируем их
13
  models = {k: TTS(k) for k in models}
 
28
  return 'temp.wav', f"Обработанный текст: '{text}'"
29
 
30
  # Создание интерфейса Gradio
31
+ model_choice = gr.Dropdown(choices=list(models.keys()), value="TeraTTS/natasha-g2p-vits", label="Модель")
32
  input_text = gr.Textbox(label="Введите текст для синтеза речи")
33
  prep_text = gr.Checkbox(label="Предобработать", info="Хотите предобработать текст? (ударения, ё)", value=True)
34
  length_scale = gr.Slider(minimum=0.1, maximum=2.0, label="Length scale (увеличить длину звучания) По умолчанию: 1.2", value=1.2)
35
 
36
  output_audio = gr.Audio(label="Аудио", type="numpy")
37
+ output_text = gr.Textbox(label="Текст")
38
 
39
  iface = gr.Interface(fn=text_to_speech, inputs=[model_choice, length_scale, input_text, prep_text], outputs=[output_audio, output_text], title=title)
40
  iface.launch()