Добавил модуль перевода числительных в цифры

sovaai · Oct 20, 2021 · aeef248 · aeef248
1 parent bb4a1bf
commit aeef248
Show file tree

Hide file tree

Showing 4 changed files with 882 additions and 1 deletion.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -30,4 +30,4 @@ services:
     image: sova-asr:master
     volumes:
       - .:/sova-asr
-    command: bash -c "python3 Train.py"
+    command: bash -c "python3 train.py"
diff --git a/file_handler.py b/file_handler.py
@@ -5,10 +5,12 @@
 import uuid
 from speech_recognizer import SpeechRecognizer
 from punctuator import Punctuator
+from number_utils.text2numbers import TextToNumbers
 
 
 speech_recognizer = SpeechRecognizer()
 punctuator = Punctuator(model_path="data/punctuator")
+text2numbers = TextToNumbers()
 
 
 class FileHandler:
@@ -55,6 +57,7 @@ def get_models_result(converted_record_path, delimiter='<br>'):
         start = time.time()
         decoder_result = speech_recognizer.recognize(converted_record_path)
         text = punctuator.predict(decoder_result.text)
+        text = text2numbers.convert(text)
         end = time.time()
         results.append(
             {