Update app.py

app.py

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