Add Dockerfile example (#828)

SYSTRAN · May 13, 2024 · 2036d12 · 2036d12
1 parent 2f6913e
commit 2036d12
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/docker/Dockerfile 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
@@ -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