Spaces:
Sleeping
Sleeping
File size: 789 Bytes
168a18b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
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)
|