chg

1 files changed, 18 insertions, 14 deletions

diff --git a/PKGBUILD b/PKGBUILD
index 720b7ef514fc..909ad8734389 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,10 +1,10 @@
-_model="small.en"
-_model_sha1sum="db8a495a91d927739e50b3fc1cc4c6b8f6c2d022"
-_pkgbase="whisper.cpp-model"
+_model='small.en'
+_model_sha1sum='db8a495a91d927739e50b3fc1cc4c6b8f6c2d022'
+_pkgbase='whisper.cpp-model'
 # Maintainer: Hauke Rehfeld <aur@haukerehfeld.de>
 pkgname="${_pkgbase}-${_model}"
 pkgver=1
-pkgrel=1
+pkgrel=2
 pkgdesc="This is an autogenerated file, please see https://github.com/hrehfeld/archlinux-whisper.cpp-model"
 arch=("i686" "x86_64")
 url="https://github.com/ggerganov/whisper.cpp/tree/master/models"
@@ -15,22 +15,26 @@ depends=()
 conflicts=()
 provides=()
 
-# copied straight from
-# https://github.com/ggerganov/whisper.cpp/blob/master/models/download-ggml-model.sh
-_url="https://huggingface.co/ggerganov/whisper.cpp"
-_pfx="resolve/main/ggml"
-_model_file="${_model}.bin"
+_model_file="ggml-${_model}.bin"
+_model_path="/usr/share/$pkgname/${_model_file}"
 
-source=("${_model_file}::${_url}/${_pfx}-${_model_file}")
-sha1sums=("${_model_sha1sum}")
+_download_script_name="download-ggml-model.sh"
+_download_script_url="https://github.com/ggerganov/whisper.cpp/raw/master/models/download-ggml-model.sh"
+
+source=("$_download_script_url")
+sha1sums=('2446b8d3ed58e79ebdd6a0272392314791fa5a3c')
+
+prepare() {
+    chmod +x $_download_script_name
+    ./$_download_script_name "$_model" "."
+}
 
 package() {
-  modelpath="/usr/share/$pkgname/${_model_file}"
-  install -Dm644 "${srcdir}/${_model_file}" "$pkgdir$modelpath"
+  install -Dm644 "${srcdir}/${_model_file}" "$pkgdir$_model_path"
 
   wrapper="whisper.cpp-${_model}"
   echo "#!/bin/sh
-/usr/bin/whisper.cpp --model ${modelpath} "\"'$@'\" > "$srcdir/$wrapper"
+/usr/bin/whisper.cpp --model ${_model_path} "\"'$@'\" > "$srcdir/$wrapper"
   install -Dm755 "${srcdir}/$wrapper" "$pkgdir/usr/bin/$wrapper"