gogs
/
rust-whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

functionize at least

master
Bel LaPointe 2024-01-03 08:28:22 -07:00
parent 5bc3209070
commit a8c8140d18
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
rust-whisper-baked-lib/src/lib.rs
View File

@ -6,7 +6,7 @@ pub fn channel<F>(
stream: std::sync::mpsc::Receiver<Vec<f32>>, stream: std::sync::mpsc::Receiver<Vec<f32>>,
) where F: FnMut(Result<rust_whisper_lib::Transcribed, String>) + Send + 'static { ) where F: FnMut(Result<rust_whisper_lib::Transcribed, String>) + Send + 'static {
flags.model_path = None; flags.model_path = None;
flags.model_buffer = Some(include_bytes!("../../models/ggml-tiny.en.bin").to_vec()); flags.model_buffer = Some(get_fast());
rust_whisper_lib::channel(flags.clone(), handler_fn, stream); rust_whisper_lib::channel(flags.clone(), handler_fn, stream);
} }
@ -15,7 +15,7 @@ pub fn wav<F>(
handler_fn: F handler_fn: F
) where F: FnMut(Result<rust_whisper_lib::Transcribed, String>) + Send + 'static { ) where F: FnMut(Result<rust_whisper_lib::Transcribed, String>) + Send + 'static {
flags.model_path = None; flags.model_path = None;
flags.model_buffer = Some(include_bytes!("../../models/ggml-distil-medium.en.bin").to_vec()); flags.model_buffer = Some(get_good());
rust_whisper_lib::wav(flags.clone(), handler_fn, flags.wav.unwrap()); rust_whisper_lib::wav(flags.clone(), handler_fn, flags.wav.unwrap());
} }
@ -24,10 +24,18 @@ pub fn wav_channel<F>(
handler_fn: F handler_fn: F
) where F: FnMut(Result<rust_whisper_lib::Transcribed, String>) + Send + 'static { ) where F: FnMut(Result<rust_whisper_lib::Transcribed, String>) + Send + 'static {
flags.model_path = None; flags.model_path = None;
flags.model_buffer = Some(include_bytes!("../../models/ggml-distil-medium.en.bin").to_vec()); flags.model_buffer = Some(get_good());
rust_whisper_lib::wav_channel(flags, handler_fn); rust_whisper_lib::wav_channel(flags, handler_fn);
} }
pub fn f32_from_wav_file(path: &String) -> Result<Vec<f32>, String> { pub fn f32_from_wav_file(path: &String) -> Result<Vec<f32>, String> {
rust_whisper_lib::f32_from_wav_file(path) rust_whisper_lib::f32_from_wav_file(path)
} }
fn get_fast() -> Vec<u8> {
include_bytes!("../../models/ggml-base.en.bin").to_vec()
}
fn get_good() -> Vec<u8> {
include_bytes!("../../models/ggml-distil-medium.en.bin").to_vec()
}