Skip to content

Commit

Permalink
Merge pull request #5 from jkrukowski/update-whisperkit-cli-name
Browse files Browse the repository at this point in the history
Update WhisperKit CLI name
  • Loading branch information
atiorh authored Mar 15, 2024
2 parents 95d178c + 2949b31 commit 3ca64f9
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion whisperkit/pipelines.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}")
Expand Down Expand Up @@ -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,
Expand Down

0 comments on commit 3ca64f9

Please sign in to comment.