Update app.py

app.py

@@ -17,7 +17,22 @@ my_inputs = [
 
 my_outputs = gr.outputs.Audio(type="file", label="Output Audio")
 
+def belarusify_russian_text(text: str):
+    text = text.replace("и", "і")
+    text = text.replace("іу", "іў")
+    text = text.replace("оу", "оў")
+    text = text.replace("ау", "аў")
+    text = text.replace("ыу", "ыў")
+    text = text.replace("уу", "уў")
+    text = text.replace("юу", "юў")
+    text = text.replace("еу", "еў")
+    text = text.replace("ёу", "ёў")
+    text = text.replace("щ", "шч")
+    return text
+    
+
 def tts(text: str):
+    text = belarusify_russian_text(text)
     best_model_path = hf_hub_download(repo_id=REPO_ID, filename="model.pth") 
     config_path = hf_hub_download(repo_id=REPO_ID, filename="config.json")
     vocoder_path = hf_hub_download(repo_id=REPO_ID, filename="vocoder.pth")