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

Default log level to DEBUG, faster-whisper to match

Browse source
This commit is contained in:
Alex Young 2024-04-17 22:21:41 +01:00
parent df64b4e2c3
commit 8060d45aea
2 changed files with 10 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
whisper_online.py
View file

@ -106,7 +106,7 @@ class FasterWhisperASR(ASRBase):
def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
from faster_whisper import WhisperModel
logging.getLogger("faster_whisper").setLevel(logging.WARNING)
logging.getLogger("faster_whisper").setLevel(logger.level)
if model_dir is not None:
logger.debug(f"Loading whisper model from model_dir {model_dir}. modelsize and cache_dir parameters are not used.")
model_size_or_path = model_dir
@ -558,6 +558,7 @@ def add_shared_args(parser):
parser.add_argument('--vad', action="store_true", default=False, help='Use VAD = voice activity detection, with the default parameters.')
parser.add_argument('--buffer_trimming', type=str, default="segment", choices=["sentence", "segment"],help='Buffer trimming strategy -- trim completed sentences marked with punctuation mark and detected by sentence segmenter, or the completed segments returned by Whisper. Sentence segmenter must be installed for "sentence" option.')
parser.add_argument('--buffer_trimming_sec', type=float, default=15, help='Buffer trimming length threshold in seconds. If buffer length is longer, trimming sentence/segment is triggered.')
parser.add_argument("-l", "--log-level", dest="log_level", choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'], help="Set the log level", default='DEBUG')
def asr_factory(args, logfile=sys.stderr):
"""
@ -624,6 +625,10 @@ if __name__ == "__main__":
logger.error("No or one option from --offline and --comp_unaware are available, not both. Exiting.")
sys.exit(1)
if args.log_level:
logging.basicConfig(format='whisper-%(levelname)s:%(name)s: %(message)s',
level=getattr(logging, args.log_level))
audio_path = args.audio_path
SAMPLING_RATE = 16000

14
whisper_online_server.py
View file

@ -8,7 +8,6 @@ import logging
import numpy as np
logger = logging.getLogger(__name__)
print(__name__)
parser = argparse.ArgumentParser()
# server options
@ -17,18 +16,13 @@ parser.add_argument("--port", type=int, default=43007)
parser.add_argument("--warmup-file", type=str, dest="warmup_file",
help="The path to a speech audio wav file to warm up Whisper so that the very first chunk processing is fast. It can be e.g. https://github.com/ggerganov/whisper.cpp/raw/master/samples/jfk.wav .")
parser.add_argument("-l", "--log-level", dest="log_level",
choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'],
help="Set the log level",
default='INFO')
# options from whisper_online
add_shared_args(parser)
args = parser.parse_args()
if args.log_level:
logging.basicConfig(format='whisper-server-%(levelname)s: %(message)s',
logging.basicConfig(format='whisper-server-%(levelname)s:%(name)s: %(message)s',
level=getattr(logging, args.log_level))
# setting whisper object by args
@ -47,11 +41,11 @@ if args.warmup_file:
if os.path.isfile(args.warmup_file):
a = load_audio_chunk(args.warmup_file,0,1)
asr.transcribe(a)
print("INFO: Whisper is warmed up.",file=sys.stderr)
logger.info("Whisper is warmed up.")
else:
print("WARNING: The warm up file is not available. "+msg,file=sys.stderr)
logger.warning("The warm up file is not available. "+msg)
else:
print("WARNING: " + msg, file=sys.stderr)
logger.warning(msg)
######### Server objects