summary refs log tree commit diff
path: root/pkgs/tools/audio/whisper-ctranslate2/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/tools/audio/whisper-ctranslate2/default.nix')
-rw-r--r--pkgs/tools/audio/whisper-ctranslate2/default.nix40
1 files changed, 40 insertions, 0 deletions
diff --git a/pkgs/tools/audio/whisper-ctranslate2/default.nix b/pkgs/tools/audio/whisper-ctranslate2/default.nix
new file mode 100644
index 00000000000..6ca3226bf67
--- /dev/null
+++ b/pkgs/tools/audio/whisper-ctranslate2/default.nix
@@ -0,0 +1,40 @@
+{ lib
+, fetchFromGitHub
+, nix-update-script
+, python3
+}:
+let
+  pname = "whisper-ctranslate2";
+  version = "0.2.7";
+in
+python3.pkgs.buildPythonApplication {
+  inherit pname version;
+  format = "setuptools";
+
+  disabled = python3.pythonOlder "3.6";
+
+  src = fetchFromGitHub {
+    owner = "Softcatala";
+    repo = pname;
+    rev = version;
+    hash = "sha256-dUmQNKgH+SIlLhUEiaEGXUHZQDr3fidsAU2vATJiXBU=";
+  };
+
+  propagatedBuildInputs = with python3.pkgs; [
+    numpy
+    faster-whisper
+    ctranslate2
+    tqdm
+    sounddevice
+  ];
+
+  passthru.updateScript = nix-update-script { };
+
+  meta = with lib; {
+    description = "Whisper command line client compatible with original OpenAI client based on CTranslate2";
+    homepage = "https://github.com/Softcatala/whisper-ctranslate2";
+    changelog = "https://github.com/Softcatala/whisper-ctranslate2/releases/tag/${version}";
+    license = licenses.mit;
+    maintainers = with maintainers; [ happysalada ];
+  };
+}