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

Fix examples

Browse source
This commit is contained in:
Niko 2025-09-29 13:40:02 -07:00
parent 31c1527abe
commit a23990c563
No known key found for this signature in database
1 changed files with 14 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

22
examples/audio_transcription.rs
View file

@ -78,6 +78,10 @@ fn main() -> Result<(), &'static str> {
.. ..
} = reader.spec(); } = reader.spec();
if sample_rate != 16000 {
panic!("sample rate must be 16KHz");
}
// Convert the audio to floating point samples. // Convert the audio to floating point samples.
let samples: Vec<i16> = reader let samples: Vec<i16> = reader
.into_samples::<i16>() .into_samples::<i16>()
@ -88,15 +92,17 @@ fn main() -> Result<(), &'static str> {
// Convert audio to 16KHz mono f32 samples, as required by the model. // Convert audio to 16KHz mono f32 samples, as required by the model.
// These utilities are provided for convenience, but can be replaced with custom conversion logic. // These utilities are provided for convenience, but can be replaced with custom conversion logic.
if channels == 2 { let audio = if channels == 1 {
audio = whisper_rs::convert_stereo_to_mono_audio(&audio).expect("Conversion error"); audio
} else if channels != 1 { } else if channels == 2 {
// be sure to initialize the output array to exactly half
// the length of the input array
let mut output = vec![0.0; audio.len() / 2];
whisper_rs::convert_stereo_to_mono_audio(&audio, &mut output).expect("Conversion error");
output
} else {
panic!(">2 channels unsupported"); panic!(">2 channels unsupported");
} };
if sample_rate != 16000 {
panic!("sample rate must be 16KHz");
}
// Run the model. // Run the model.
state.full(params, &audio[..]).expect("failed to run model"); state.full(params, &audio[..]).expect("failed to run model");