yumoqing/fastwhisper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2025-11-06 15:00:02 +08:00
parent c1803a0ede
commit 8236dd18a9
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
fw/fw.py
View File

@ -41,17 +41,21 @@ class MyLongTask(LongTasks):
def transcribe(self, model, fpath): def transcribe(self, model, fpath):
segments, info = model.transcribe(fpath, beam_size=5) segments, info = model.transcribe(fpath, beam_size=5)
debug(f'{segments=}, {info=}')
segments = list(segments) segments = list(segments)
debug(f'{segments=}, {info=}')
words = []
for s in segments:
if s.words:
words.append([word.start, word.end, word.word for word in s.words])
return { return {
'language': info.language, 'language': info.language,
'language_probability': info.language_probability, 'language_probability': info.language_probability,
'content': ' '.join([s.text for s in segments]), 'content': ' '.join([s.text for s in segments]),
'words': words,
'segments': [[segment.start, 'segments': [[segment.start,
segment.end, segment.end,
segment.text, segment.text
[[word.start, word.end, word.word] for word in segment.words ]
] for segment in segments] ] for segment in segments]
} }