heimoshuiyu/whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

format code

Browse Source
This commit is contained in:
黑墨水鱼
2023-04-20 02:02:11 +08:00
parent 67fac2a4ce
commit 62e34f0c03
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
whisper/transcribe.py
View File

@@ -361,12 +361,17 @@ def transcribe(
)
for i, segment in enumerate(current_segments, start=len(all_segments)):
if not segment['text'].strip():
if not segment["text"].strip():
continue
check_prompt_num = 1
if any([prev['text'].strip() == segment['text'].strip() for prev in all_prompts_segments[-check_prompt_num:]]):
if any(
[
prev["text"].strip() == segment["text"].strip()
for prev in all_prompts_segments[-check_prompt_num:]
]
):
continue
all_prompts_tokens.extend(segment['tokens'])
all_prompts_tokens.extend(segment["tokens"])
all_prompts_segments.append({"id": i, **segment})
# update progress bar