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

add logger for online asr

Browse source
This commit is contained in:
Silas Kieser 2025-01-21 14:45:45 +01:00
parent 25eb276794
commit 69a2ed6bfb
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
whisper_online.py
View file

@ -236,10 +236,12 @@ def asr_factory(args, logfile=sys.stderr):
online = online_factory(args, asr, tokenizer, logfile=logfile) online = online_factory(args, asr, tokenizer, logfile=logfile)
return asr, online return asr, online
def set_logging(args, logger, other="_server"): def set_logging(args, logger, others=[]):
logging.basicConfig(format="%(levelname)s\t%(message)s") # format='%(name)s logging.basicConfig(format="%(levelname)s\t%(message)s") # format='%(name)s
logger.setLevel(args.log_level) logger.setLevel(args.log_level)
logging.getLogger("whisper_online" + other).setLevel(args.log_level)
for other in others:
logging.getLogger(other).setLevel(args.log_level)
# logging.getLogger("whisper_online_server").setLevel(args.log_level) # logging.getLogger("whisper_online_server").setLevel(args.log_level)
@ -276,7 +278,7 @@ if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
# reset to store stderr to different file stream, e.g. open(os.devnull,"w") # reset to store stderr to different file stream, e.g. open(os.devnull,"w")
logfile = sys.stderr logfile = None # sys.stderr
if args.offline and args.comp_unaware: if args.offline and args.comp_unaware:
logger.error( logger.error(
@ -288,7 +290,7 @@ if __name__ == "__main__":
# logging.basicConfig(format='whisper-%(levelname)s:%(name)s: %(message)s', # logging.basicConfig(format='whisper-%(levelname)s:%(name)s: %(message)s',
# level=getattr(logging, args.log_level)) # level=getattr(logging, args.log_level))
set_logging(args, logger) set_logging(args, logger,others=["src.whisper_streaming.online_asr"])
audio_path = args.audio_path audio_path = args.audio_path