| 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") | |