daihui.zhang commited on
Commit
3ec4a4f
·
1 Parent(s): 730ea7e

fix save wav file error

Browse files
Files changed (1) hide show
  1. transcribe/utils.py +2 -1
transcribe/utils.py CHANGED
@@ -93,7 +93,8 @@ def resample(file: str, sr: int = 16000):
93
 
94
 
95
  def save_to_wave(filename, data:np.ndarray, sample_rate=16000):
96
- write(filename, sample_rate, data.astype(np.int16))
 
97
 
98
 
99
  def pcm_bytes_to_np_array(pcm_bytes: bytes, dtype=np.float32, channels=1):
 
93
 
94
 
95
  def save_to_wave(filename, data:np.ndarray, sample_rate=16000):
96
+ data = (data * 32767).astype(np.int16)
97
+ write(filename, sample_rate, data)
98
 
99
 
100
  def pcm_bytes_to_np_array(pcm_bytes: bytes, dtype=np.float32, channels=1):