lightmare/whispercpp.py
1
1
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update whispercpp.pyx

Browse Source
This commit is contained in:
Luke Southam 2022-12-16 01:11:18 +00:00 committed by GitHub
parent c5f51d5b55
commit 64008a7d64
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
whispercpp.pyx
View File

@ -103,7 +103,8 @@ cdef class Whisper:
cdef audio_data data = load_audio(<bytes>filename) cdef audio_data data = load_audio(<bytes>filename)
return whisper_full(self.ctx, self.params, data.frames, data.n_frames) return whisper_full(self.ctx, self.params, data.frames, data.n_frames)
cpdef list extract_text(self, int res): def extract_text(self, int res):
print("Extracting text...")
if res != 0: if res != 0:
raise RuntimeError raise RuntimeError
cdef int n_segments = whisper_full_n_segments(self.ctx) cdef int n_segments = whisper_full_n_segments(self.ctx)