Update infer_onnx.py

infer_onnx.py

@@ -56,8 +56,14 @@ class TTS:
         phoneme_ids_inter = self._intersperse(phoneme_ids, 0)
         return phoneme_ids_inter
         
+    def _num2wordsshor(self, match):
+        match = match.group()
+        ret = num2words(match, lang ='ru')
+        return ret 
     
     def __call__(self, text: str, length_scale=1.2):
+        text = translit(text, 'ru')
+        text = re.sub(r'\d+',self._num2wordsshor,text)
         phoneme_ids = self._get_seq(text)
         text = np.expand_dims(np.array(phoneme_ids, dtype=np.int64), 0)
         text_lengths = np.array([text.shape[1]], dtype=np.int64)