Add text_preformatter

app.py

@@ -6,6 +6,7 @@ from TTS.utils.manage import ModelManager
 from TTS.utils.synthesizer import Synthesizer
 import requests
 from os.path import exists
+from formatter import preprocess_text
 
 MODEL_NAMES = [
     "uk/mai/vits-tts"
@@ -45,6 +46,7 @@ for MODEL_NAME in MODEL_NAMES:
 
 def tts(text: str, model_name: str):
     text = text if len(text) < 500 else text[0:500] # mitigate crashes on hf space
+    text = preprocess_text(text)
     print(text, model_name)
     synthesizer = MODELS.get(model_name, None)
     if synthesizer is None:

formatter.py

@@ -0,0 +1,16 @@
+def preprocess_text(text):
+    # replace apostrophe
+    text = text.replace("`", "'")
+    text = text.replace("ʼ", "'")
+    # numbers
+    text = text.replace("1", "один ")
+    text = text.replace("2", "два ")
+    text = text.replace("3", "три ")
+    text = text.replace("4", "чотири ")
+    text = text.replace("5", "п'ять ")
+    text = text.replace("6", "шість ")
+    text = text.replace("7", "сім ")
+    text = text.replace("8", "вісім ")
+    text = text.replace("9", "дев'ять ")
+    text = text.replace("0", "нуль ")
+    return text

requirements.txt

@@ -1 +1,2 @@
-TTS==0.4.1
+TTS==0.4.1
+gradio