chg

2 files changed, 21 insertions, 17 deletions

diff --git a/.SRCINFO b/.SRCINFO
index cb89073bcbce..70a1cd849971 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,12 +1,12 @@
 pkgbase = whisper.cpp-model-base.en
 	pkgdesc = This is an autogenerated file, please see https://github.com/hrehfeld/archlinux-whisper.cpp-model
 	pkgver = 1
-	pkgrel = 1
+	pkgrel = 2
 	url = https://github.com/ggerganov/whisper.cpp/tree/master/models
 	arch = i686
 	arch = x86_64
 	license = MIT
-	source = base.en.bin::https://huggingface.co/ggerganov/whisper.cpp/resolve/main/ggml-base.en.bin
-	sha1sums = 137c40403d78fd54d454da0f9bd998f78703390c
+	source = https://github.com/ggerganov/whisper.cpp/raw/master/models/download-ggml-model.sh
+	sha1sums = 2446b8d3ed58e79ebdd6a0272392314791fa5a3c
 
 pkgname = whisper.cpp-model-base.en
diff --git a/PKGBUILD b/PKGBUILD
index fdb47317338b..9635dad42ae7 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,10 +1,10 @@
-_model="base.en"
-_model_sha1sum="137c40403d78fd54d454da0f9bd998f78703390c"
-_pkgbase="whisper.cpp-model"
+_model='base.en'
+_model_sha1sum='137c40403d78fd54d454da0f9bd998f78703390c'
+_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"