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

delimiters

Browse Source
This commit is contained in:
bel
2023-04-08 22:22:22 -06:00
parent 0aff4f556b
commit 7c369e72d4
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
rust-whisper.d/hotwords.py
View File

@@ -142,9 +142,13 @@ class Parser(threading.Thread):
log("stderr:", proc.stderr.decode().strip()) log("stderr:", proc.stderr.decode().strip())
log("raw transcript:", result) log("raw transcript:", result)
result = result.replace(">>", "") result = result.replace(">>", "")
result = "".join([i.split("]")[-1] for i in result.split("[")[0]]) for pair in [
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(pair[1])[-1] for i in result.split(pair[0])[0]])
if os.environ.get("DEBUG", None): if os.environ.get("DEBUG", None):
log("annotation-free transcript:", result) log("annotation-free transcript:", result)
return result return result