Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
from infer_onnx import TTS
|
3 |
+
|
4 |
+
natasha = TTS("TeraTTS/natasha-vits")
|
5 |
+
glados = TTS("TeraTTS/glados-vits")
|
6 |
+
|
7 |
+
|
8 |
+
def text_to_speech(model_name, text):
|
9 |
+
|
10 |
+
if model_name == "TeraTTS/natasha-vits":
|
11 |
+
audio = natasha(text)
|
12 |
+
natasha.save_wav(audio, 'temp.wav')
|
13 |
+
|
14 |
+
if model_name == "TeraTTS/glados-vits":
|
15 |
+
audio = glados(text)
|
16 |
+
glados.save_wav(audio, 'temp.wav')
|
17 |
+
|
18 |
+
return 'temp.wav'
|
19 |
+
|
20 |
+
input_text = gr.Textbox(label="Введите текст для синтеза речи")
|
21 |
+
model_choice = gr.Dropdown(choices=["TeraTTS/natasha-vits", "TeraTTS/glados-vits"], value="TeraTTS/natasha-vits", label="Выберите модель")
|
22 |
+
output_audio = gr.Audio(label="Аудио", type="numpy")
|
23 |
+
|
24 |
+
iface = gr.Interface(fn=text_to_speech, inputs=[model_choice, input_text], outputs=output_audio,
|
25 |
+
description="## [Ссылка](https://github.com/Tera2Space/RUTTS/) на наш гитхаб с кодом для локального запуска"))
|
26 |
+
iface.launch()
|