LLM/WhisperLiveKit
1
0
Fork 0
mirror of https://github.com/QuentinFuxa/WhisperLiveKit.git synced 2026-05-01 19:37:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Turn prints into logging.debug calls in whisper_online_server.py

Browse Source
This commit is contained in:
Alex Young
2024-04-14 14:24:59 +01:00
parent 1f2352fa1d
commit 5ebbed3bd7
1 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
whisper_online_server.py
View File

@@ -4,17 +4,27 @@ from whisper_online import *
import sys import sys
import argparse import argparse
import os import os
import logging
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
# server options # server options
parser.add_argument("--host", type=str, default='localhost') parser.add_argument("--host", type=str, default='localhost')
parser.add_argument("--port", type=int, default=43007) parser.add_argument("--port", type=int, default=43007)
parser.add_argument("-l", "--log-level", dest="log_level",
choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'],
help="Set the log level",
default='INFO')
# options from whisper_online # options from whisper_online
add_shared_args(parser) add_shared_args(parser)
args = parser.parse_args() args = parser.parse_args()
if args.log_level:
logging.basicConfig(format='whisper-server-%(levelname)s: %(message)s',
level=getattr(logging, args.log_level))
# setting whisper object by args # setting whisper object by args
@@ -24,7 +34,7 @@ size = args.model
language = args.lan language = args.lan
t = time.time() t = time.time()
print(f"Loading Whisper {size} model for {language}...",file=sys.stderr,end=" ",flush=True) logging.debug(f"Loading Whisper {size} model for {language}...")
if args.backend == "faster-whisper": if args.backend == "faster-whisper":
from faster_whisper import WhisperModel from faster_whisper import WhisperModel
@@ -44,10 +54,10 @@ else:
tgt_language = language tgt_language = language
e = time.time() e = time.time()
print(f"done. It took {round(e-t,2)} seconds.",file=sys.stderr) logging.debug(f"done. It took {round(e-t,2)} seconds.")
if args.vad: if args.vad:
print("setting VAD filter",file=sys.stderr) logging.debug("setting VAD filter")
asr.use_vad() asr.use_vad()
@@ -70,9 +80,7 @@ if os.path.exists(demo_audio_path):
# warm up the ASR, because the very first transcribe takes much more time than the other # warm up the ASR, because the very first transcribe takes much more time than the other
asr.transcribe(a) asr.transcribe(a)
else: else:
print("Whisper is not warmed up",file=sys.stderr) logging.info("Whisper is not warmed up")
######### Server objects ######### Server objects
@@ -80,9 +88,6 @@ else:
import line_packet import line_packet
import socket import socket
import logging
class Connection: class Connection:
'''it wraps conn object''' '''it wraps conn object'''
PACKET_SIZE = 65536 PACKET_SIZE = 65536
@@ -191,11 +196,6 @@ class ServerProcessor:
# Start logging.
level = logging.INFO
logging.basicConfig(level=level, format='whisper-server-%(levelname)s: %(message)s')
# server loop # server loop
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s: with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s: