#!/usr/bin/env bash
set -euo pipefail

mkdir -p models

curl -L -o models/whisper_encoder_base_20s.onnx \
  https://ftrg.zbox.filez.com/v2/delivery/data/95f00b0fc900458ba134f8b180b3f7a1/examples/whisper/whisper_encoder_base_20s.onnx

curl -L -o models/whisper_decoder_base_20s.onnx \
  https://ftrg.zbox.filez.com/v2/delivery/data/95f00b0fc900458ba134f8b180b3f7a1/examples/whisper/whisper_decoder_base_20s.onnx

curl -L -o models/mel_80_filters.txt \
  https://raw.githubusercontent.com/airockchip/rknn_model_zoo/master/examples/whisper/model/mel_80_filters.txt

curl -L -o models/vocab_en.txt \
  https://raw.githubusercontent.com/airockchip/rknn_model_zoo/master/examples/whisper/model/vocab_en.txt

curl -L -o models/vocab_zh.txt \
  https://raw.githubusercontent.com/airockchip/rknn_model_zoo/master/examples/whisper/model/vocab_zh.txt

echo "Downloaded Whisper model assets to ./models"