diff --git a/docker/Dockerfile b/docker/Dockerfile new file mode 100644 index 0000000..b4d6aa7 --- /dev/null +++ b/docker/Dockerfile @@ -0,0 +1,8 @@ +FROM nvidia/cuda:12.0.0-runtime-ubuntu20.04 + +WORKDIR /root +RUN apt-get update -y && apt-get install -y python3 python3-pip libcudnn8 libcudnn8-dev +COPY infer.py jfk.flac ./ +RUN pip3 install faster-whisper + +ENTRYPOINT ["python3", "infer.py"] diff --git a/docker/infer.py b/docker/infer.py new file mode 100644 index 0000000..5d6b12c --- /dev/null +++ b/docker/infer.py @@ -0,0 +1,7 @@ +from faster_whisper import WhisperModel + +jfk_path = "jfk.flac" +model = WhisperModel("tiny", device="cuda") +segments, info = model.transcribe(jfk_path, word_timestamps=True) +for segment in segments: + print("[%.2fs -> %.2fs] %s" % (segment.start, segment.end, segment.text)) diff --git a/docker/jfk.flac b/docker/jfk.flac new file mode 100644 index 0000000..e44b7c1 Binary files /dev/null and b/docker/jfk.flac differ