

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