heimoshuiyu/whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use stdout for printing transcription progress (#867)

Browse Source
This commit is contained in:
Jong Wook Kim
2023-01-20 00:54:05 -08:00
committed by GitHub
parent ea1c266709
commit 12e1089462
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
whisper/transcribe.py
View File

@@ -169,7 +169,8 @@ def transcribe(
line = f"[{format_timestamp(start)} --> {format_timestamp(end)}] {text}\n" line = f"[{format_timestamp(start)} --> {format_timestamp(end)}] {text}\n"
# compared to just `print(line)`, this replaces any character not representable using # compared to just `print(line)`, this replaces any character not representable using
# the system default encoding with an '?', avoiding UnicodeEncodeError. # the system default encoding with an '?', avoiding UnicodeEncodeError.
sys.stderr.buffer.write(line.encode(sys.getdefaultencoding(), errors="replace")) sys.stdout.buffer.write(line.encode(sys.getdefaultencoding(), errors="replace"))
sys.stdout.flush()
# show the progress bar when verbose is False (otherwise the transcribed text will be printed) # show the progress bar when verbose is False (otherwise the transcribed text will be printed)
num_frames = mel.shape[-1] num_frames = mel.shape[-1]