Commit
·
dd2f0f0
1
Parent(s):
615a8be
add logger for online asr
Browse files- whisper_online.py +6 -4
whisper_online.py
CHANGED
@@ -236,10 +236,12 @@ def asr_factory(args, logfile=sys.stderr):
|
|
236 |
online = online_factory(args, asr, tokenizer, logfile=logfile)
|
237 |
return asr, online
|
238 |
|
239 |
-
def set_logging(args, logger,
|
240 |
logging.basicConfig(format="%(levelname)s\t%(message)s") # format='%(name)s
|
241 |
logger.setLevel(args.log_level)
|
242 |
-
|
|
|
|
|
243 |
|
244 |
|
245 |
# logging.getLogger("whisper_online_server").setLevel(args.log_level)
|
@@ -276,7 +278,7 @@ if __name__ == "__main__":
|
|
276 |
args = parser.parse_args()
|
277 |
|
278 |
# reset to store stderr to different file stream, e.g. open(os.devnull,"w")
|
279 |
-
logfile = sys.stderr
|
280 |
|
281 |
if args.offline and args.comp_unaware:
|
282 |
logger.error(
|
@@ -288,7 +290,7 @@ if __name__ == "__main__":
|
|
288 |
# logging.basicConfig(format='whisper-%(levelname)s:%(name)s: %(message)s',
|
289 |
# level=getattr(logging, args.log_level))
|
290 |
|
291 |
-
set_logging(args, logger)
|
292 |
|
293 |
audio_path = args.audio_path
|
294 |
|
|
|
236 |
online = online_factory(args, asr, tokenizer, logfile=logfile)
|
237 |
return asr, online
|
238 |
|
239 |
+
def set_logging(args, logger, others=[]):
|
240 |
logging.basicConfig(format="%(levelname)s\t%(message)s") # format='%(name)s
|
241 |
logger.setLevel(args.log_level)
|
242 |
+
|
243 |
+
for other in others:
|
244 |
+
logging.getLogger(other).setLevel(args.log_level)
|
245 |
|
246 |
|
247 |
# logging.getLogger("whisper_online_server").setLevel(args.log_level)
|
|
|
278 |
args = parser.parse_args()
|
279 |
|
280 |
# reset to store stderr to different file stream, e.g. open(os.devnull,"w")
|
281 |
+
logfile = None # sys.stderr
|
282 |
|
283 |
if args.offline and args.comp_unaware:
|
284 |
logger.error(
|
|
|
290 |
# logging.basicConfig(format='whisper-%(levelname)s:%(name)s: %(message)s',
|
291 |
# level=getattr(logging, args.log_level))
|
292 |
|
293 |
+
set_logging(args, logger,others=["src.whisper_streaming.online_asr"])
|
294 |
|
295 |
audio_path = args.audio_path
|
296 |
|