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

Add a --warmup-file option to pass in a path

Browse source
This commit is contained in:
Alex Young 2024-04-14 19:29:46 +01:00
parent 1f2352fa1d
commit 70bc57180c
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
whisper_online_server.py
View file

@ -10,6 +10,8 @@ parser = argparse.ArgumentParser()
parser.add_argument("--host", type=str, default='localhost') parser.add_argument("--host", type=str, default='localhost')
parser.add_argument("--port", type=int, default=43007) parser.add_argument("--port", type=int, default=43007)
parser.add_argument("--warmup-file", type=str, dest="warmup_file")
# options from whisper_online # options from whisper_online
add_shared_args(parser) add_shared_args(parser)
@ -61,10 +63,9 @@ online = OnlineASRProcessor(asr,tokenizer,buffer_trimming=(args.buffer_trimming,
demo_audio_path = "cs-maji-2.16k.wav" if os.path.exists(args.warmup_file):
if os.path.exists(demo_audio_path):
# load the audio into the LRU cache before we start the timer # load the audio into the LRU cache before we start the timer
a = load_audio_chunk(demo_audio_path,0,1) a = load_audio_chunk(args.warmup_file,0,1)
# TODO: it should be tested whether it's meaningful # TODO: it should be tested whether it's meaningful
# warm up the ASR, because the very first transcribe takes much more time than the other # warm up the ASR, because the very first transcribe takes much more time than the other