admin-valentin/WhisperLiveKit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

specify audio dtype

Browse source
This commit is contained in:
koiking213 2024-02-20 22:46:04 +09:00
parent db8b7d2883
commit 24926c98e0
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
whisper_online.py
View file

@ -11,7 +11,7 @@ import math
@lru_cache @lru_cache
def load_audio(fname): def load_audio(fname):
a, _ = librosa.load(fname, sr=16000) a, _ = librosa.load(fname, sr=16000)
return a return a.astype('float32')
def load_audio_chunk(fname, beg, end): def load_audio_chunk(fname, beg, end):
audio = load_audio(fname) audio = load_audio(fname)

2
whisper_online_server.py
View file

@ -138,7 +138,7 @@ class ServerProcessor:
break break
sf = soundfile.SoundFile(io.BytesIO(raw_bytes), channels=1,endian="LITTLE",samplerate=SAMPLING_RATE, subtype="PCM_16",format="RAW") sf = soundfile.SoundFile(io.BytesIO(raw_bytes), channels=1,endian="LITTLE",samplerate=SAMPLING_RATE, subtype="PCM_16",format="RAW")
audio, _ = librosa.load(sf,sr=SAMPLING_RATE) audio, _ = librosa.load(sf,sr=SAMPLING_RATE)
out.append(audio) out.append(audio.astype('float32'))
if not out: if not out:
return None return None
return np.concatenate(out) return np.concatenate(out)