20 lines
783 B
Python
20 lines
783 B
Python
import whisper
|
||
|
||
model = whisper.load_model("turbo", device="cuda")
|
||
result = model.transcribe("audio2.mp4", word_timestamps=False, initial_prompt="这是小学四年级习作问题诊断及指导的视频。", verbose=False, language='zh')
|
||
|
||
# result = model.transcribe("audio.mp3", word_timestamps=False, initial_prompt="这是一段小说。", verbose=False, language='zh')
|
||
|
||
|
||
def seconds_to_min_sec(total_seconds: int) -> str:
|
||
minutes, seconds = divmod(total_seconds, 60)
|
||
return f"{minutes}分{seconds:02d}秒"
|
||
|
||
|
||
for item in result['segments']:
|
||
print('时间：' + str(seconds_to_min_sec(int(item['start']))) + ' ~ ' + str(
|
||
seconds_to_min_sec(int(item['end']))) + '；' + '文本：' + str(
|
||
item['text']))
|
||
|
||
print('所有文本内容：' + result['text'])
|