regularfry commited on
Commit
8060d45
·
1 Parent(s): df64b4e

Default log level to DEBUG, faster-whisper to match

Browse files
Files changed (2) hide show
  1. whisper_online.py +6 -1
  2. whisper_online_server.py +4 -10
whisper_online.py CHANGED
@@ -106,7 +106,7 @@ class FasterWhisperASR(ASRBase):
106
 
107
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
108
  from faster_whisper import WhisperModel
109
- logging.getLogger("faster_whisper").setLevel(logging.WARNING)
110
  if model_dir is not None:
111
  logger.debug(f"Loading whisper model from model_dir {model_dir}. modelsize and cache_dir parameters are not used.")
112
  model_size_or_path = model_dir
@@ -558,6 +558,7 @@ def add_shared_args(parser):
558
  parser.add_argument('--vad', action="store_true", default=False, help='Use VAD = voice activity detection, with the default parameters.')
559
  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.')
560
  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.')
 
561
 
562
  def asr_factory(args, logfile=sys.stderr):
563
  """
@@ -624,6 +625,10 @@ if __name__ == "__main__":
624
  logger.error("No or one option from --offline and --comp_unaware are available, not both. Exiting.")
625
  sys.exit(1)
626
 
 
 
 
 
627
  audio_path = args.audio_path
628
 
629
  SAMPLING_RATE = 16000
 
106
 
107
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
108
  from faster_whisper import WhisperModel
109
+ logging.getLogger("faster_whisper").setLevel(logger.level)
110
  if model_dir is not None:
111
  logger.debug(f"Loading whisper model from model_dir {model_dir}. modelsize and cache_dir parameters are not used.")
112
  model_size_or_path = model_dir
 
558
  parser.add_argument('--vad', action="store_true", default=False, help='Use VAD = voice activity detection, with the default parameters.')
559
  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.')
560
  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.')
561
+ parser.add_argument("-l", "--log-level", dest="log_level", choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'], help="Set the log level", default='DEBUG')
562
 
563
  def asr_factory(args, logfile=sys.stderr):
564
  """
 
625
  logger.error("No or one option from --offline and --comp_unaware are available, not both. Exiting.")
626
  sys.exit(1)
627
 
628
+ if args.log_level:
629
+ logging.basicConfig(format='whisper-%(levelname)s:%(name)s: %(message)s',
630
+ level=getattr(logging, args.log_level))
631
+
632
  audio_path = args.audio_path
633
 
634
  SAMPLING_RATE = 16000
whisper_online_server.py CHANGED
@@ -8,7 +8,6 @@ import logging
8
  import numpy as np
9
 
10
  logger = logging.getLogger(__name__)
11
- print(__name__)
12
  parser = argparse.ArgumentParser()
13
 
14
  # server options
@@ -17,18 +16,13 @@ parser.add_argument("--port", type=int, default=43007)
17
  parser.add_argument("--warmup-file", type=str, dest="warmup_file",
18
  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 .")
19
 
20
- parser.add_argument("-l", "--log-level", dest="log_level",
21
- choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'],
22
- help="Set the log level",
23
- default='INFO')
24
-
25
 
26
  # options from whisper_online
27
  add_shared_args(parser)
28
  args = parser.parse_args()
29
 
30
  if args.log_level:
31
- logging.basicConfig(format='whisper-server-%(levelname)s: %(message)s',
32
  level=getattr(logging, args.log_level))
33
 
34
  # setting whisper object by args
@@ -47,11 +41,11 @@ if args.warmup_file:
47
  if os.path.isfile(args.warmup_file):
48
  a = load_audio_chunk(args.warmup_file,0,1)
49
  asr.transcribe(a)
50
- print("INFO: Whisper is warmed up.",file=sys.stderr)
51
  else:
52
- print("WARNING: The warm up file is not available. "+msg,file=sys.stderr)
53
  else:
54
- print("WARNING: " + msg, file=sys.stderr)
55
 
56
 
57
  ######### Server objects
 
8
  import numpy as np
9
 
10
  logger = logging.getLogger(__name__)
 
11
  parser = argparse.ArgumentParser()
12
 
13
  # server options
 
16
  parser.add_argument("--warmup-file", type=str, dest="warmup_file",
17
  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 .")
18
 
 
 
 
 
 
19
 
20
  # options from whisper_online
21
  add_shared_args(parser)
22
  args = parser.parse_args()
23
 
24
  if args.log_level:
25
+ logging.basicConfig(format='whisper-server-%(levelname)s:%(name)s: %(message)s',
26
  level=getattr(logging, args.log_level))
27
 
28
  # setting whisper object by args
 
41
  if os.path.isfile(args.warmup_file):
42
  a = load_audio_chunk(args.warmup_file,0,1)
43
  asr.transcribe(a)
44
+ logger.info("Whisper is warmed up.")
45
  else:
46
+ logger.warning("The warm up file is not available. "+msg)
47
  else:
48
+ logger.warning(msg)
49
 
50
 
51
  ######### Server objects