# Copyright 2026 Trunkmasters
# Distributed under the terms of the GNU General Public License v2

EAPI=8

# NVIDIA's Parakeet TDT 0.6B v2 (English speech recognition), the ONNX export of istupakov at a
# pinned revision: NeMo's 128-bin mel features, the full precision encoder (its weights in the
# .data file beside it; the 8-bit encoder is slower on CPUs without fast integer kernels) and the
# decoder/joint network in both precisions. Data only; www-apps/attendant's attendant-speech runs
# it on sci-libs/onnxruntime. The version is the export revision's date (2026-02-17).
HF_REV="0bbb45a3365852604aef28b538a8f066f4ccaa85"
HF_URI="https://huggingface.co/istupakov/parakeet-tdt-0.6b-v2-onnx/resolve/${HF_REV}"
FILES="nemo128.onnx encoder-model.onnx encoder-model.onnx.data decoder_joint-model.onnx
	decoder_joint-model.int8.onnx vocab.txt config.json README.md"

DESCRIPTION="NVIDIA Parakeet TDT 0.6B v2 speech-to-text model (ONNX)"
HOMEPAGE="
	https://huggingface.co/nvidia/parakeet-tdt-0.6b-v2
	https://huggingface.co/istupakov/parakeet-tdt-0.6b-v2-onnx
"
SRC_URI=""
for f in ${FILES}; do
	SRC_URI+=" ${HF_URI}/${f} -> ${P}-${f}"
done
S="${WORKDIR}"

LICENSE="CC-BY-4.0"
SLOT="0"
KEYWORDS="~amd64"
RESTRICT="mirror strip"

src_unpack() { :; }

src_install() {
	local f
	insinto /usr/share/parakeet-tdt-0.6b-v2
	for f in ${FILES}; do
		newins "${DISTDIR}/${P}-${f}" "${f}"
	done
}