gogs
/
rust-whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

debug and model via env

master
bel 2023-03-29 19:07:16 -06:00
parent 0e278fffcb
commit f3bb3431c9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
whisper-2023/hotwords.py
View File

@ -112,6 +112,8 @@ class Parser(threading.Thread):
log("Parser.run: start") log("Parser.run: start")
while True: while True:
try: try:
if environ.get("DEBUG", None):
log(self.inq.qsize())
clip = self.inq.get() clip = self.inq.get()
if clip is None: if clip is None:
break break
@ -123,7 +125,7 @@ class Parser(threading.Thread):
def _run(self, clip): def _run(self, clip):
r = sr.Recognizer() r = sr.Recognizer()
return r.recognize_whisper(clip, language="english", model="small.en") # tiny.en=32x, base.en=16x, small.en=6x return r.recognize_whisper(clip, language="english", model=environ.get("MODEL", "small.en")) # tiny.en=32x, base.en=16x, small.en=6x, medium.en=x2
def load_dot_notation(v, s): def load_dot_notation(v, s):
items = s.replace("[]", ".[]").split(".") items = s.replace("[]", ".[]").split(".")