Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions app/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,14 +20,14 @@ class Settings(BaseSettings):

silero_settings = {
'ru': {
'model_link': 'https://models.silero.ai/models/tts/ru/v4_ru.pt',
'model_link': 'https://models.silero.ai/models/tts/ru/v5_1_ru.pt',
'model_name': 'ru_model.pt',
'speakers': [
'aidar', 'baya', 'kseniya', 'xenia', 'eugene', 'random'
]
},
'uk': {
'model_link': 'https://models.silero.ai/models/tts/ua/v3_ua.pt',
'model_link': 'https://models.silero.ai/models/tts/ua/v4_ua.pt',
'model_name': 'uk_model.pt',
'speakers': [
'mykyta', 'random'
Expand Down
4 changes: 2 additions & 2 deletions app/normalizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,13 +19,13 @@ def normalize_time(text: str) -> str:


def normalize_number(text: str) -> str:
number_strings = findall(r'(?<![a-zA-Z\d])\d+(?:\.\d+)?(?:(?:\s|\w)*?<d>.*?</d>)*(?!(?:[a-zA-Z\d\"\']|\s)*\'?/?>)',
number_strings = findall(r'(?<![a-zA-Z\d])\d+(?:\,\d+)?(?:(?:\s|\w)*?<d>.*?</d>)*(?!(?:[a-zA-Z\d\"\']|\s)*\'?/?>)',
text)

for number_string in number_strings:
number_data = number_string.split(' ')

number = num2words(number_data[0], lang=settings.language)
number = num2words(number_data[0].replace(',', '.'), lang=settings.language)
number_gender = None

inflected_words = []
Expand Down
2 changes: 0 additions & 2 deletions dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,6 @@ ENV PYTHONUNBUFFERED=1
RUN python3 -m ensurepip
RUN pip3 install --no-cache --upgrade pip setuptools

RUN pip3 install torch>=2.0.1+cpu -f https://download.pytorch.org/whl/torch_stable.html

ADD requirements_docker.txt .
RUN pip3 install -r requirements_docker.txt

Expand Down
3 changes: 2 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -7,4 +7,5 @@ numpy~=1.23.5
pymorphy2~=0.9.1
pymorphy2-dicts-uk~=2.4.1.1.1460299261
transliterate~=1.10.2
requests~=2.28.1
requests~=2.28.1
scipy~=1.14.0
17 changes: 10 additions & 7 deletions requirements_docker.txt
Original file line number Diff line number Diff line change
@@ -1,9 +1,12 @@
uvicorn~=0.20.0
fastapi~=0.88.0
pydantic~=1.10.2
num2words~=0.5.12
numpy~=1.23.5
--extra-index-url https://download.pytorch.org/whl/cpu
uvicorn~=0.38.0
fastapi~=0.121.3
pydantic<2.0.0
num2words~=0.5.14
numpy~=2.2.6
pymorphy2~=0.9.1
pymorphy2-dicts-uk~=2.4.1.1.1460299261
pymorphy2-dicts-ru~=2.4.417127.4579844
transliterate~=1.10.2
requests~=2.28.1
requests~=2.32.5
scipy~=1.15.3
torch==2.9.1+cpu