From 0aff4f556b766d8046defe45eb0e5540fef3d837 Mon Sep 17 00:00:00 2001 From: bel Date: Sat, 8 Apr 2023 20:05:03 -0600 Subject: [PATCH] one more --- rust-whisper.d/hotwords.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rust-whisper.d/hotwords.py b/rust-whisper.d/hotwords.py index 769d8bd..bf548b6 100644 --- a/rust-whisper.d/hotwords.py +++ b/rust-whisper.d/hotwords.py @@ -136,7 +136,7 @@ class Parser(threading.Thread): p = "/tmp/whisper-cpp.wav" with open("/tmp/whisper-cpp.wav", "wb") as f: f.write(wav) - proc = subprocess.run(f"MODEL=./models/ggml-{os.environ.get('MODEL','tiny.en')}.bin WAV={p} P=2 rust-whisper", capture_output=True, shell=True) + proc = subprocess.run(f"MODEL=./models/ggml-{os.environ.get('MODEL','tiny.en')}.bin WAV={p} P={os.environ.get('P', '2')} rust-whisper", capture_output=True, shell=True) result = proc.stdout.decode().strip() if os.environ.get("DEBUG", None): log("stderr:", proc.stderr.decode().strip()) @@ -144,6 +144,7 @@ class Parser(threading.Thread): result = result.replace(">>", "") result = "".join([i.split("]")[-1] for i in result.split("[")[0]]) result = "".join([i.split(")")[-1] for i in result.split("(")[0]]) + result = "".join([i.split(">")[-1] for i in result.split("<")[0]]) if os.environ.get("DEBUG", None): log("annotation-free transcript:", result) return result