cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove flake8 warning suppression

Browse Source 
Signed-off-by: makaveli10 <suryanvineet47@gmail.com>
This commit is contained in:
makaveli10
2024-05-28 21:29:38 +05:30
parent 399e9e7efe
commit 3d043dc906
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/transcriber.py
+5 -5
View File
@@ -423,7 +423,7 @@ class WhisperModel:
return segments, info
def generate_segments( # noqa: C901
def generate_segments(
self,
features: np.ndarray,
tokenizer: Tokenizer,
@@ -496,7 +496,7 @@ class WhisperModel:
content_frames - seek,
seek_clip_end - seek,
)
segment = features[:, seek : seek + segment_size] # noqa: E203
segment = features[:, seek:seek + segment_size]
segment_duration = segment_size * self.feature_extractor.time_per_frame
segment = pad_or_trim(segment, self.feature_extractor.nb_max_frames)
@@ -685,7 +685,7 @@ class WhisperModel:
continue
if is_segment_anomaly(segment):
next_segment = next_words_segment(
current_segments[si + 1 :] # noqa: E203
current_segments[si + 1:]
)
if next_segment is not None:
hal_next_start = next_segment["words"][0]["start"]
@@ -909,7 +909,7 @@ class WhisperModel:
if previous_tokens:
prompt.append(tokenizer.sot_prev)
prompt.extend(previous_tokens[-(self.max_length // 2 - 1) :]) # noqa: E203
prompt.extend(previous_tokens[-(self.max_length // 2 - 1):])
prompt.extend(tokenizer.sot_sequence)
@@ -926,7 +926,7 @@ class WhisperModel:
return prompt
def add_word_timestamps( # noqa: C901
def add_word_timestamps(
self,
segments: List[dict],
tokenizer: Tokenizer,