Spaces:
Running
Running
Yurii Paniv
commited on
Commit
•
12ee3cc
1
Parent(s):
de21232
Apply black formatter
Browse files- .gitignore +3 -0
- .vscode/settings.json +3 -0
- app.py +4 -3
- formatter.py +3 -1
- stress_with_model.py +1 -4
.gitignore
CHANGED
@@ -128,6 +128,9 @@ dmypy.json
|
|
128 |
# Pyre type checker
|
129 |
.pyre/
|
130 |
|
|
|
|
|
|
|
131 |
# model files
|
132 |
*.pth.tar
|
133 |
*.pth
|
|
|
128 |
# Pyre type checker
|
129 |
.pyre/
|
130 |
|
131 |
+
# mac
|
132 |
+
.DS_Store
|
133 |
+
|
134 |
# model files
|
135 |
*.pth.tar
|
136 |
*.pth
|
.vscode/settings.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"python.formatting.provider": "black"
|
3 |
+
}
|
app.py
CHANGED
@@ -67,7 +67,9 @@ def tts(text: str, voice: str, stress: str):
|
|
67 |
print("Voice", voice)
|
68 |
print("Stress:", stress)
|
69 |
print("Time:", datetime.utcnow())
|
70 |
-
autostress_with_model =
|
|
|
|
|
71 |
speaker_name = "male1" if voice == VoiceOption.MaleVoice.value else "female3"
|
72 |
text = preprocess_text(text, autostress_with_model)
|
73 |
text_limit = 7200
|
@@ -93,7 +95,7 @@ iface = gr.Interface(
|
|
93 |
gr.inputs.Radio(
|
94 |
label="Голос",
|
95 |
choices=[option.value for option in VoiceOption],
|
96 |
-
default=VoiceOption.FemaleVoice.value
|
97 |
),
|
98 |
gr.inputs.Radio(
|
99 |
label="Наголоси",
|
@@ -105,7 +107,6 @@ iface = gr.Interface(
|
|
105 |
gr.outputs.Textbox(label="Наголошений текст"),
|
106 |
],
|
107 |
title="🐸💬🇺🇦 - Coqui TTS",
|
108 |
-
#theme="huggingface",
|
109 |
description="Україномовний🇺🇦 TTS за допомогою Coqui TTS (щоб вручну поставити наголос, використовуйте + перед голосною)",
|
110 |
article="Якщо вам подобається, підтримайте за посиланням: [SUPPORT LINK](https://send.monobank.ua/jar/48iHq4xAXm), "
|
111 |
+ "Github: [https://github.com/robinhad/ukrainian-tts](https://github.com/robinhad/ukrainian-tts) \n"
|
|
|
67 |
print("Voice", voice)
|
68 |
print("Stress:", stress)
|
69 |
print("Time:", datetime.utcnow())
|
70 |
+
autostress_with_model = (
|
71 |
+
True if stress == StressOption.AutomaticStressWithModel.value else False
|
72 |
+
)
|
73 |
speaker_name = "male1" if voice == VoiceOption.MaleVoice.value else "female3"
|
74 |
text = preprocess_text(text, autostress_with_model)
|
75 |
text_limit = 7200
|
|
|
95 |
gr.inputs.Radio(
|
96 |
label="Голос",
|
97 |
choices=[option.value for option in VoiceOption],
|
98 |
+
default=VoiceOption.FemaleVoice.value,
|
99 |
),
|
100 |
gr.inputs.Radio(
|
101 |
label="Наголоси",
|
|
|
107 |
gr.outputs.Textbox(label="Наголошений текст"),
|
108 |
],
|
109 |
title="🐸💬🇺🇦 - Coqui TTS",
|
|
|
110 |
description="Україномовний🇺🇦 TTS за допомогою Coqui TTS (щоб вручну поставити наголос, використовуйте + перед голосною)",
|
111 |
article="Якщо вам подобається, підтримайте за посиланням: [SUPPORT LINK](https://send.monobank.ua/jar/48iHq4xAXm), "
|
112 |
+ "Github: [https://github.com/robinhad/ukrainian-tts](https://github.com/robinhad/ukrainian-tts) \n"
|
formatter.py
CHANGED
@@ -78,7 +78,9 @@ def preprocess_text(text, use_autostress_model=False):
|
|
78 |
text = text.replace(english_char.upper(), english[english_char].upper())
|
79 |
text = text.replace(english_char, english[english_char])
|
80 |
|
81 |
-
text = sentence_to_stress(
|
|
|
|
|
82 |
|
83 |
return text
|
84 |
|
|
|
78 |
text = text.replace(english_char.upper(), english[english_char].upper())
|
79 |
text = text.replace(english_char, english[english_char])
|
80 |
|
81 |
+
text = sentence_to_stress(
|
82 |
+
text, stress_with_model if use_autostress_model else stress_dict
|
83 |
+
)
|
84 |
|
85 |
return text
|
86 |
|
stress_with_model.py
CHANGED
@@ -3,10 +3,7 @@ import ukrainian_accentor as accentor
|
|
3 |
# run
|
4 |
def stress_with_model(text: str):
|
5 |
text = text.lower()
|
6 |
-
|
7 |
-
result = accentor.process(text, mode='plus')
|
8 |
-
except ValueError: # TODO: apply fix for cases when there are no vowels
|
9 |
-
return text
|
10 |
return result
|
11 |
|
12 |
|
|
|
3 |
# run
|
4 |
def stress_with_model(text: str):
|
5 |
text = text.lower()
|
6 |
+
result = accentor.process(text, mode="plus")
|
|
|
|
|
|
|
7 |
return result
|
8 |
|
9 |
|