Update infer_onnx.py

infer_onnx.py

@@ -4,7 +4,7 @@ import onnxruntime
 import numpy as np
 from huggingface_hub import snapshot_download
 from gruut import sentences
-import time
+import numpy as np
 import scipy.io.wavfile
 
 
@@ -66,6 +66,12 @@ class TTS:
         text_norm = self._intersperse(text_norm, 0)
         return text_norm
 
+    def _add_silent(self, audio, silence_duration: float = 0.7, sample_rate: int = 22050):
+        num_samples_silence = int(sample_rate * silence_duration)
+        silence_array = np.zeros(num_samples_silence, dtype=np.float32)
+        audio_with_silence = np.concatenate((audio, silence_array), axis=0)
+        return audio_with_silence
+
     def save_wav(self, audio, path:str):
         '''save audio to wav'''
         scipy.io.wavfile.write(path, 22050, audio)
@@ -87,5 +93,5 @@ class TTS:
                 "sid": None,
             },
         )[0][0,0][0]
-        
+        audio = self._add_silent(audio)
         return audio