From f04a55590fd782159d040b5018fc0e29e848c4d7 Mon Sep 17 00:00:00 2001 From: Bel LaPointe <153096461+breel-render@users.noreply.github.com> Date: Sat, 21 Sep 2024 17:37:50 -0400 Subject: [PATCH] fixed --- wav_to_mkv.d/wav_subtitles.rs | 2 +- wav_to_mkv.d/wav_to_mkv.sh | 9 ++++++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/wav_to_mkv.d/wav_subtitles.rs b/wav_to_mkv.d/wav_subtitles.rs index f4a59b2..2695be8 100644 --- a/wav_to_mkv.d/wav_subtitles.rs +++ b/wav_to_mkv.d/wav_subtitles.rs @@ -22,7 +22,7 @@ fn main() -> Result<(), &'static str> { // Edit params as needed. // Set the number of threads to use to 1. - params.set_n_threads(1); + //params.set_n_threads(1); // Enable translation. params.set_translate(true); // Set the language to translate to to English. diff --git a/wav_to_mkv.d/wav_to_mkv.sh b/wav_to_mkv.d/wav_to_mkv.sh index b65570c..be5e30b 100644 --- a/wav_to_mkv.d/wav_to_mkv.sh +++ b/wav_to_mkv.d/wav_to_mkv.sh @@ -3,17 +3,20 @@ main() { set -euo pipefail - input_wav="$1" - model="${2:-../models/ggml-small.en.bin}" + input_wav="$(realpath "$1")" + model="$(realpath "${2:-../models/ggml-small.en.bin}")" already_transcribed="${3:-false}" sanitized_wav="${input_wav%.*}.mono-16khz.wav" ffmpeg -y -i "$input_wav" -ac 1 -ar 16k "$sanitized_wav" if ! $already_transcribed; then + pushd "$(dirname "$(realpath "$BASH_SOURCE")")" + cd ../gitea-whisper-rs/ cargo run --example wav_subtitles -- "$model" "$sanitized_wav" + popd fi - out_to_srt ./transcript.txt > "${input_wav%.*}.srt" + out_to_srt ../gitea-whisper-rs/transcript.txt > "${input_wav%.*}.srt" ffmpeg -y -i "$input_wav" -i "${input_wav%.*}.srt" "${input_wav%.*}.mkv" ls "${input_wav%.*}.mkv"