|
import torch |
|
import gradio as gr |
|
|
|
from infer import inference |
|
|
|
|
|
|
|
description = f''' |
|
Програма може не коректно визначати деякі наголоси і не перетворює цифри, акроніми і різні скорочення в словесну форму. |
|
Якщо наголос не правильний, використовуйте символ + після наголошеного складу. |
|
Також дуже маленькі речення можуть крешати, тому пишіть щось більше а не одне-два слова. |
|
|
|
''' |
|
|
|
def synthesise(text, speed, progress=gr.Progress()): |
|
if text.strip() == "": |
|
raise gr.Error("You must enter some text") |
|
if len(text) > 150000: |
|
raise gr.Error("Text must be <150k characters") |
|
print("*** saying ***") |
|
print(text) |
|
print("*** end ***") |
|
|
|
return 24000, inference(text, progress, speed=speed, alpha=1.0, diffusion_steps=6, embedding_scale=1.0)[0] |
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
i = gr.Interface( |
|
fn=synthesise, |
|
description=description, |
|
inputs=[ |
|
gr.Text(label='Text:', lines=5, max_lines=10), |
|
gr.Slider(label='Швидкість:', maximum=1.3, minimum=0.7, value=1.0), |
|
], |
|
outputs=[ |
|
gr.Audio( |
|
label="Audio:", |
|
autoplay=False, |
|
streaming=False, |
|
type="numpy", |
|
), |
|
|
|
], |
|
allow_flagging ='never', |
|
cache_examples=False, |
|
title='StyleTTS2 ukrainian demo', |
|
examples=[["""Решта окупантів звернула на Вокзальну — центральну вулицю Бучі. Тільки уявіть їхній настрій, коли перед ними відкрилася ця пасторальна картина! Невеличкі котеджі й просторіші будинки шикуються обабіч, перед ними вивищуються голі липи та електростовпи, тягнуться газони й жовто-чорні бордюри. Доглянуті сади визирають із-поза зелених парканів, гавкотять собаки, співають птахи… На дверях будинку номер тридцять шість досі висить різдвяний вінок.""", 1.0], [""" |
|
Одна дівчинка стала королевою Франції. Звали її Анна, і була вона донькою Ярослава Му+дрого, великого київського князя. Він опі+кувався літературою та культурою в Київській Русі+, а тоді переважно про таке не дбали – більше воювали і споруджували фортеці.""", 1.0], [""" |
|
Одна дівчинка народилася і виросла в Америці, та коли стала дорослою, зрозуміла, що дуже любить українські вірші й найбільше хоче робити вистави про Україну. Звали її Вірляна. Дід Вірляни був український мовознавець і педагог Кость Кисілевський, котрий навчався в Лейпцизькому та Віденському університетах і, після Другої світової війни виїхавши до США, започаткував систему шкіл українознавства по всій Америці. Тож Вірляна зростала в українському середовищі, а окрім того – в середовищі вихідців з інших країн.""", 1.0] ], |
|
) |
|
i.queue(max_size=20, default_concurrency_limit=4) |
|
i.launch(share=False, server_name="0.0.0.0") |
|
|