SilasKieser commited on
Commit
dd2f0f0
·
1 Parent(s): 615a8be

add logger for online asr

Browse files
Files changed (1) hide show
  1. 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, other="_server"):
240
  logging.basicConfig(format="%(levelname)s\t%(message)s") # format='%(name)s
241
  logger.setLevel(args.log_level)
242
- logging.getLogger("whisper_online" + other).setLevel(args.log_level)
 
 
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