Update app.py
Browse files
app.py
CHANGED
@@ -2,11 +2,13 @@ import gradio as gr
|
|
2 |
from infer_onnx import TTS
|
3 |
from ruaccent import RUAccent # https://huggingface.co/TeraTTS/accentuator
|
4 |
|
|
|
|
|
5 |
models = ["TeraTTS/natasha-g2p-vits", "TeraTTS/glados2-g2p-vits"]
|
6 |
|
7 |
models = {k:TTS(k) for k in models}
|
8 |
|
9 |
-
accentizer = RUAccent(workdir="./model/ruaccent")
|
10 |
accentizer.load(omograph_model_size='medium', dict_load_startup=True)
|
11 |
|
12 |
|
@@ -29,5 +31,7 @@ prep_text = gr.Checkbox(label="Предобработать", info="Хотите
|
|
29 |
output_audio = gr.Audio(label="Аудио", type="numpy")
|
30 |
output_text = gr.Textbox(label="Обработанный текст")
|
31 |
|
32 |
-
iface = gr.Interface(fn=text_to_speech, inputs=[model_choice, input_text, prep_text], outputs=[output_audio, output_text])
|
33 |
-
iface.launch()
|
|
|
|
|
|
2 |
from infer_onnx import TTS
|
3 |
from ruaccent import RUAccent # https://huggingface.co/TeraTTS/accentuator
|
4 |
|
5 |
+
title = "GitHub with models: https://github.com/Tera2Space/RUTTS"
|
6 |
+
|
7 |
models = ["TeraTTS/natasha-g2p-vits", "TeraTTS/glados2-g2p-vits"]
|
8 |
|
9 |
models = {k:TTS(k) for k in models}
|
10 |
|
11 |
+
accentizer = RUAccent(workdir="./model/ruaccent", allow_cuda=False)
|
12 |
accentizer.load(omograph_model_size='medium', dict_load_startup=True)
|
13 |
|
14 |
|
|
|
31 |
output_audio = gr.Audio(label="Аудио", type="numpy")
|
32 |
output_text = gr.Textbox(label="Обработанный текст")
|
33 |
|
34 |
+
iface = gr.Interface(fn=text_to_speech, inputs=[model_choice, input_text, prep_text], outputs=[output_audio, output_text], title=title)
|
35 |
+
iface.launch()
|
36 |
+
|
37 |
+
|