Skip to main content

CLI for Omi Med STT v1 medical speech-to-text

Project description

Omi Med STT Runtime

PyPI Tests License: MIT

Command-line runtime for Omi Med STT v1, an English medical speech-to-text model built from NVIDIA Parakeet TDT 0.6B v2.

The package downloads the right model artifact for your machine and transcribes audio locally.

Install

pip install -U omi-med-stt

Apple Silicon:

pip install -U "omi-med-stt[mlx]"

NVIDIA CUDA / NeMo:

pip install -U "omi-med-stt[nemo]"

Run

omi-med-stt audio.wav

Useful options:

omi-med-stt audio.wav --json
omi-med-stt audio.wav --runtime mlx
omi-med-stt audio.wav --runtime nemo
omi-med-stt audio.wav --runtime cpp
omi-med-stt check

Audio formats. WAV, FLAC, OGG and other libsndfile formats are read directly. Other inputs — .m4a (iPhone Voice Memos / QuickTime), .mp3, .aac, .mp4, .mov, .wma, .opus, .webm, … — are decoded with ffmpeg, which ships with the package, so there's nothing extra to install. If a system ffmpeg is on your PATH it's used instead (e.g. a newer build). Whatever the input, audio is downmixed to mono and resampled to 16 kHz automatically.

Runtime Choices

Platform Default runtime Model artifact
Apple Silicon mlx omi-health/omi-med-stt-v1-mlx-q8
NVIDIA CUDA nemo omi-health/omi-med-stt-v1
Linux/Windows CPU cpp omi-health/omi-med-stt-v1-gguf

The canonical model is the NeMo checkpoint. MLX and GGUF are runtime exports.

CPU setup:

omi-med-stt install-cpp --cpp-backend cpu
omi-med-stt audio.wav --runtime cpp

The CPU path uses a patched parakeet.cpp runtime and downloads the q8_0 GGUF artifact only. It does not download the NeMo or MLX weights.

Runtime Quality

Artifact WER M-WER Drug M-WER Medical Recall
NeMo canonical 8.30% 2.37% 4.75% 97.95%
MLX full precision 8.59% 2.65% 5.20% 97.70%
MLX q8 8.61% 2.75% 5.20% 97.63%
GGUF q8_0 9.12% 3.20% 6.33% 97.53%

These numbers compare the released runtime artifacts against each other on the same internal benchmark. Visit omi.health for the broader model evaluation and product context.

Model Repositories

If the model repositories are private before launch, authenticate first:

huggingface-cli login

CUDA Note

If --runtime nemo fails with a CUDA driver mismatch, install a PyTorch wheel matching your driver before installing the NeMo extra. For example, on CUDA 12.8 hosts:

pip install torch --index-url https://download.pytorch.org/whl/cu128
pip install -U "omi-med-stt[nemo]"

Development

git clone https://github.com/Omi-Health/omi-med-stt-runtime
cd omi-med-stt-runtime
pip install -e ".[dev]"
python scripts/prepublish_check.py --skip-build
python -m pytest -q tests

Safety

Omi Med STT v1 is speech-to-text only. It is not a diagnostic, triage, prescribing, or clinical decision model, and it is not clinically validated. Transcripts must be reviewed before any clinical use.

License And Attribution

Runtime code is MIT licensed.

Model weights are CC-BY-4.0 and are derived from nvidia/parakeet-tdt-0.6b-v2. Omi Med STT v1 is not an NVIDIA model.

The CPU runtime uses parakeet.cpp.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

omi_med_stt-0.1.21.tar.gz (37.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

omi_med_stt-0.1.21-py3-none-any.whl (27.2 kB view details)

Uploaded Python 3

File details

Details for the file omi_med_stt-0.1.21.tar.gz.

File metadata

  • Download URL: omi_med_stt-0.1.21.tar.gz
  • Upload date:
  • Size: 37.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.9

File hashes

Hashes for omi_med_stt-0.1.21.tar.gz
Algorithm Hash digest
SHA256 a4222d516ed10180bc10e03947a89c5580c2291a2411abad9c036c655d5ce63b
MD5 8e61ada5721bfb062cf6d3d972e49144
BLAKE2b-256 6b0c1472fb86f2dba3607ba9d5fb89bd86abe5ade2428f3fbfddec41efeeba3d

See more details on using hashes here.

File details

Details for the file omi_med_stt-0.1.21-py3-none-any.whl.

File metadata

  • Download URL: omi_med_stt-0.1.21-py3-none-any.whl
  • Upload date:
  • Size: 27.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.9

File hashes

Hashes for omi_med_stt-0.1.21-py3-none-any.whl
Algorithm Hash digest
SHA256 0bd2c288aabe016e0b3df57f29edaabf1712b14da970077c2b43db92aadebc9a
MD5 398a0733d30c37e7f5b7d6986cebdffe
BLAKE2b-256 a08ff3129634fc4e30a168f613539120d128f6e6113031cd5e8c5ada2cf47fe8

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page