Python script to use Whisper to transcribe audio/video

translate.py

import os
import whisper
from whisper.utils import get_writer

input_dir = 'input/'
output_dir = 'output/'

for filename in sorted(os.listdir(input_dir)):
    if filename.endswith('.mp4'):
        input_file = input_dir + filename
        print('Processing ' + input_file + '...')
        model = whisper.load_model('large')
        result = model.transcribe(input_file, fp16=False)
        srt_writer = get_writer('srt', output_dir)
        srt_writer(result, input_file)