video_translator / transcriptFromWhisper.py
Gotenks1893's picture
Upload 8 files
168a18b
raw
history blame contribute delete
789 Bytes
import whisper
import os
def getAllWhisperTranscripts():
directory = os.fsencode("videos")
for file in os.listdir(directory):
filename = os.fsdecode(file)
print(filename)
model = whisper.load_model("tiny")
result = model.transcribe("videos/" + filename, fp16=False)
fname = "whisper/" + filename
fname = fname.replace(".mp4", ".txt")
with open(fname, "w") as f:
f.write(result["text"])
print("written to: " + fname)
def getWhisperTranscript(videoId):
model = whisper.load_model("tiny")
result = model.transcribe(f"videos/{videoId}.mp4", fp16=False)
fname = f"whisper/{videoId}.txt"
with open(fname, "w") as f:
f.write(result["text"])
print("written to: " + fname)