update whisperkit cli name

argmaxinc · Mar 15, 2024 · 2949b31 · 2949b31
1 parent 95d178c
commit 2949b31
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/whisperkit/pipelines.py b/whisperkit/pipelines.py
@@ -115,7 +115,7 @@ def clone_repo(self):
             commit_hash=self.code_commit_hash)
 
     def build_cli(self):
-        self.product_name = "transcribe"
+        self.product_name = "whisperkit-cli"
         if subprocess.check_call(f"swift build -c release --product {self.product_name}",
                                  cwd=self.repo_dir, shell=True):
             raise subprocess.CalledProcessError(f"Failed to build {self.product_name}")
@@ -181,6 +181,7 @@ def transcribe(self, audio_file_path: str) -> str:
 
         cmd = " ".join([
             self.cli_path,
+            "transcribe",
             "--audio-path", audio_file_path,
             "--model-path", self.models_dir,
             "--text-decoder-compute-units", self.compute_unit.value,