Spaces:
Sleeping
Sleeping
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) | |