harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add a rustified progress callback.

Browse source
This commit is contained in:
Marcin Mielniczuk 2023-08-11 15:08:05 +02:00
parent 19f47dac39
commit ddabeb4c0b
2 changed files with 42 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
examples/full_usage/src/main.rs
View file

@ -47,7 +47,8 @@ fn main() {
let ctx = WhisperContext::new(&whisper_path.to_string_lossy()).expect("failed to open model");
let mut state = ctx.create_state().expect("failed to create key");
let params = FullParams::new(SamplingStrategy::default());
let mut params = FullParams::new(SamplingStrategy::default());
params.set_progress_callback_safe(|progress| println!("Progress callback: {}%", progress));
let st = std::time::Instant::now();
state