Commit @c83746b5ab261f93bfd48d939a11664eb074a99c - yjyoon/whisper_streaming

Alex Young 2024-04-14

Turn prints into logging.debug calls in whisper_online_server.py

@c83746b5ab261f93bfd48d939a11664eb074a99c

c27a20e

c83746b

whisper_online_server.py

--- whisper_online_server.py

+++ whisper_online_server.py


 import sys
 import argparse
 import os
+import logging
+
 parser = argparse.ArgumentParser()
 
 # server options
 parser.add_argument("--host", type=str, default='localhost')
 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
 add_shared_args(parser)
 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 
 

 language = args.lan
 
 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":
     from faster_whisper import WhisperModel

     tgt_language = language
 
 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:
-    print("setting VAD filter",file=sys.stderr)
+    logging.debug("setting VAD filter")
     asr.use_vad()
 
 

     # warm up the ASR, because the very first transcribe takes much more time than the other
     asr.transcribe(a)
 else:
-    print("Whisper is not warmed up",file=sys.stderr)
-
-
+    logging.info("Whisper is not warmed up")
 
 
 ######### Server objects
 
 import line_packet
 import socket
-
-import logging
-
 
 class Connection:
     '''it wraps conn object'''

 #        self.send_result(o)
 
 
-
-
-# Start logging.
-level = logging.INFO
-logging.basicConfig(level=level, format='whisper-server-%(levelname)s: %(message)s')
 
 # server loop
 

Add a comment

Open 0
Closed 0

List

...	...	@@ -4,17 +4,27 @@
4	4	import sys
5	5	import argparse
6	6	import os
	7	+import logging
	8	+
7	9	parser = argparse.ArgumentParser()
8	10
9	11	# server options
10	12	parser.add_argument("--host", type=str, default='localhost')
11	13	parser.add_argument("--port", type=int, default=43007)
12	14
	15	+parser.add_argument("-l", "--log-level", dest="log_level",
	16	+ choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'],
	17	+ help="Set the log level",
	18	+ default='INFO')
	19	+
13	20
14	21	# options from whisper_online
15	22	add_shared_args(parser)
16	23	args = parser.parse_args()
17	24
	25	+if args.log_level:
	26	+ logging.basicConfig(format='whisper-server-%(levelname)s: %(message)s',
	27	+ level=getattr(logging, args.log_level))
18	28
19	29	# setting whisper object by args
20	30
...	...	@@ -24,7 +34,7 @@
24	34	language = args.lan
25	35
26	36	t = time.time()
27		-print(f"Loading Whisper {size} model for {language}...",file=sys.stderr,end=" ",flush=True)
	37	+logging.debug(f"Loading Whisper {size} model for {language}...")
28	38
29	39	if args.backend == "faster-whisper":
30	40	from faster_whisper import WhisperModel
...	...	@@ -44,10 +54,10 @@
44	54	tgt_language = language
45	55
46	56	e = time.time()
47		-print(f"done. It took {round(e-t,2)} seconds.",file=sys.stderr)
	57	+logging.debug(f"done. It took {round(e-t,2)} seconds.")
48	58
49	59	if args.vad:
50		- print("setting VAD filter",file=sys.stderr)
	60	+ logging.debug("setting VAD filter")
51	61	asr.use_vad()
52	62
53	63
...	...	@@ -70,18 +80,13 @@
70	80	# warm up the ASR, because the very first transcribe takes much more time than the other
71	81	asr.transcribe(a)
72	82	else:
73		- print("Whisper is not warmed up",file=sys.stderr)
74		-
75		-
	83	+ logging.info("Whisper is not warmed up")
76	84
77	85
78	86	######### Server objects
79	87
80	88	import line_packet
81	89	import socket
82		-
83		-import logging
84		-
85	90
86	91	class Connection:
87	92	'''it wraps conn object'''
...	...	@@ -190,11 +195,6 @@
190	195	# self.send_result(o)
191	196
192	197
193		-
194		-
195		-# Start logging.
196		-level = logging.INFO
197		-logging.basicConfig(level=level, format='whisper-server-%(levelname)s: %(message)s')
198	198
199	199	# server loop
200	200

Delete comment