diff --git a/.gitignore b/.gitignore index 31726f3..367bf96 100644 --- a/.gitignore +++ b/.gitignore @@ -1,5 +1,7 @@ **/*.sw* /whisper-cpp-2023/rust.d/target +/rust-whisper.d/target +/rust-whisper.d/models snowboy-2022/snowboy **/*.git.d **/*.wav diff --git a/whisper-cpp-2023/rust.d/Cargo.lock b/rust-whisper.d/Cargo.lock similarity index 100% rename from whisper-cpp-2023/rust.d/Cargo.lock rename to rust-whisper.d/Cargo.lock diff --git a/whisper-cpp-2023/rust.d/Cargo.toml b/rust-whisper.d/Cargo.toml similarity index 100% rename from whisper-cpp-2023/rust.d/Cargo.toml rename to rust-whisper.d/Cargo.toml diff --git a/whisper-cpp-2023/rust.d/cargo.sh b/rust-whisper.d/cargo.sh similarity index 100% rename from whisper-cpp-2023/rust.d/cargo.sh rename to rust-whisper.d/cargo.sh diff --git a/rust-whisper.d/download_models.sh b/rust-whisper.d/download_models.sh new file mode 100755 index 0000000..dba5e7e --- /dev/null +++ b/rust-whisper.d/download_models.sh @@ -0,0 +1,10 @@ +#!/bin/bash + +src="https://huggingface.co/ggerganov/whisper.cpp/resolve/main/ggml" +d="${1:-"$PWD"/models}" +mkdir -p "$d" + +# Whisper models +for model in "tiny.en" "base.en" "small.en" "medium.en"; do + test -f "$d"/ggml-$model.bin || wget --quiet --show-progress -O "$d"/ggml-$model.bin "$src-$model.bin" +done diff --git a/whisper-cpp-2023/hotwords.py b/rust-whisper.d/hotwords.py similarity index 100% rename from whisper-cpp-2023/hotwords.py rename to rust-whisper.d/hotwords.py diff --git a/whisper-cpp-2023/rust.d/src/main.rs b/rust-whisper.d/src/main.rs similarity index 100% rename from whisper-cpp-2023/rust.d/src/main.rs rename to rust-whisper.d/src/main.rs diff --git a/whisper-cpp-2023/native.sh b/whisper-cpp-2023/native.sh index da82c6f..fa92531 100644 --- a/whisper-cpp-2023/native.sh +++ b/whisper-cpp-2023/native.sh @@ -5,7 +5,7 @@ if [ ! -d ./git.d/.git ]; then fi cd ./git.d -if [ ! -f ./samples/gb1.wav ]; then +if [ ! -f ./samples/jfk.wav ]; then make samples fi if [ ! -f ./main ]; then @@ -20,5 +20,5 @@ fi if [ -n "$STREAM" ]; then ./stream -m ./models/ggml-${MODEL:-tiny.en}.bin -t 8 --step 500 --length ${MIC_TIMEOUT:-2}000 $(test -n "$MIC_ID" && echo -c "$MIC_ID") else - time ./main -m ./models/ggml-${MODEL:-tiny.en}.bin -f ./samples/gb1.wav -t 4 + time ./main -m ./models/ggml-${MODEL:-tiny.en}.bin -f ./samples/jfk.wav -t 4 fi