diff --git a/examples/basic_use.rs b/examples/basic_use.rs index 53d2aab..54e7d5d 100644 --- a/examples/basic_use.rs +++ b/examples/basic_use.rs @@ -51,15 +51,16 @@ fn main() { // note that you don't need to use these, you can do it yourself or any other way you want // these are just provided for convenience let mut inter_samples = vec![Default::default(); samples.len()]; + let mut mono_samples = vec![Default::default(); samples.len() / 2]; whisper_rs::convert_integer_to_float_audio(&samples, &mut inter_samples) .expect("failed to convert audio data"); - let samples = whisper_rs::convert_stereo_to_mono_audio(&inter_samples) + whisper_rs::convert_stereo_to_mono_audio(&inter_samples, &mut mono_samples) .expect("failed to convert audio data"); // now we can run the model state - .full(params, &samples[..]) + .full(params, &mono_samples[..]) .expect("failed to run model"); // fetch the results diff --git a/src/utilities.rs b/src/utilities.rs index 2e11b7c..cf30539 100644 --- a/src/utilities.rs +++ b/src/utilities.rs @@ -51,7 +51,8 @@ pub fn convert_integer_to_float_audio( /// ``` /// # use whisper_rs::convert_stereo_to_mono_audio; /// let samples = [0.0f32; 1024]; -/// let mono = convert_stereo_to_mono_audio(&samples).expect("should be no half samples missing"); +/// let mut mono_samples = [0.0f32; 512]; +/// convert_stereo_to_mono_audio(&samples, &mut mono_samples).expect("should be no half samples missing"); /// ``` pub fn convert_stereo_to_mono_audio(input: &[f32], output: &mut [f32]) -> Result<(), WhisperError> { let (input, []) = input.as_chunks::<2>() else { diff --git a/src/whisper_params.rs b/src/whisper_params.rs index b51b602..846af3b 100644 --- a/src/whisper_params.rs +++ b/src/whisper_params.rs @@ -801,7 +801,7 @@ impl<'a, 'b> FullParams<'a, 'b> { /// # Examples /// ``` /// # use whisper_rs::{FullParams, SamplingStrategy}; - /// let mut params = FullParams::new(SamplingStrategy::default()); + /// let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 5 }); /// params.set_initial_prompt("Hello, world!"); /// // ... further usage of params ... /// ```