diff --git a/rust-whisper.d/Cargo.lock b/rust-whisper.d/Cargo.lock index 6054c46..52f9276 100644 --- a/rust-whisper.d/Cargo.lock +++ b/rust-whisper.d/Cargo.lock @@ -261,21 +261,20 @@ dependencies = [ [[package]] name = "whisper-rs" -version = "0.8.0" +version = "0.5.0" source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "2c950fb18ad556b053ba615b88fd4d01ed6020be740c3371eb0fc4aec64a0639" +checksum = "fa7e1b9b003aa3285a0e4469219566266aa1d51ced1be38587251a4f713a1677" dependencies = [ "whisper-rs-sys", ] [[package]] name = "whisper-rs-sys" -version = "0.6.1" +version = "0.3.1" source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "094a5bd86f6f52562bbc74c28f27cd80197e54656cfb7213cf4ba37b5246cc9e" +checksum = "97a389dc665c7354ba6b1982850d4ba05b862907e535708ebdec92cbd9c599e8" dependencies = [ "bindgen", - "cfg-if", ] [[package]] diff --git a/rust-whisper.d/Cargo.toml b/rust-whisper.d/Cargo.toml index 8d22ae4..62b4532 100644 --- a/rust-whisper.d/Cargo.toml +++ b/rust-whisper.d/Cargo.toml @@ -6,6 +6,6 @@ edition = "2021" # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html [dependencies] -whisper-rs = "0.8.0" +whisper-rs = "0.5.0" wav = "1" tokio = "1.27" diff --git a/rust-whisper.d/src/main.rs b/rust-whisper.d/src/main.rs index d7cdbec..618b640 100644 --- a/rust-whisper.d/src/main.rs +++ b/rust-whisper.d/src/main.rs @@ -1,10 +1,10 @@ use whisper_rs::{WhisperContext, FullParams, SamplingStrategy}; fn main() { - let ctx = WhisperContext::new( + let mut ctx = WhisperContext::new( &std::env::var("MODEL").unwrap_or(String::from("../models/ggml-tiny.en.bin")) ).expect("failed to load model"); - let mut state = ctx.create_state().expect("failed to create state"); + //let mut state = ctx.create_state().expect("failed to create state"); // create a params object let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 0 }); @@ -26,11 +26,11 @@ fn main() { let data16 = data.as_sixteen().expect("wav is not 32bit floats"); let audio_data = &whisper_rs::convert_integer_to_float_audio(&data16); - state.full(params, &audio_data[..]).expect("failed to run model"); + ctx.full(params, &audio_data[..]).expect("failed to run model"); - let num_segments = state.full_n_segments().expect("failed to get number of segments"); + let num_segments = ctx.full_n_segments(); //.expect("failed to get number of segments"); for i in 0..num_segments { - let segment = state.full_get_segment_text(i).expect("failed to get segment"); + let segment = ctx.full_get_segment_text(i).expect("failed to get segment"); print!("{} ", segment); } println!("");