Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 06:26:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for ffmpeg return code false positive errors

Browse Source
This commit is contained in:
Raivis Dejus 2024-10-04 11:24:13 +03:00
parent 25639fc17d
commit 76db75f5b5
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
whisper/audio.py
View File

@ -51,15 +51,16 @@ def load_audio(file: str, sr: int = SAMPLE_RATE):
"-ac", "1",
"-acodec", "pcm_s16le",
"-ar", str(sr),
"-loglevel", "error",
"-"
]
# fmt: on
try:
out = run(cmd, capture_output=True, check=True).stdout
except CalledProcessError as e:
raise RuntimeError(f"Failed to load audio: {e.stderr.decode()}") from e
result = run(cmd, capture_output=True)
return np.frombuffer(out, np.int16).flatten().astype(np.float32) / 32768.0
if len(result.stderr):
raise RuntimeError(f"Failed to load audio: {result.stderr.decode()}")
return np.frombuffer(result.stdout, np.int16).flatten().astype(np.float32) / 32768.0
def pad_or_trim(array, length: int = N_SAMPLES, *, axis: int = -1):