Merge pull request #630 from mlopsengr/patch-1

Update README.md
This commit is contained in:
Max Bain
2023-12-17 15:53:44 +00:00
committed by GitHub

View File

@ -165,6 +165,10 @@ compute_type = "float16" # change to "int8" if low on GPU mem (may reduce accura
# 1. Transcribe with original whisper (batched)
model = whisperx.load_model("large-v2", device, compute_type=compute_type)
# save model to local path (optional)
# model_dir = "/path/"
# model = whisperx.load_model("large-v2", device, compute_type=compute_type, download_root=model_dir)
audio = whisperx.load_audio(audio_file)
result = model.transcribe(audio, batch_size=batch_size)
print(result["segments"]) # before alignment