vosk 0.3.1

Safe wrapper around the Vosk API Speech Recognition Toolkit
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

use std::{ffi::CString, ptr::NonNull};
use vosk_sys::*;

/// The same as [`Model`], but uses
/// a CUDA enabled Nvidia GPU and dynamic batching to enable higher throughput.
///
/// [`Model`]: super::sequential::Model
pub struct BatchModel(pub(crate) NonNull<VoskBatchModel>);

impl BatchModel {
    /// Loads model data from the file and returns the model object, or [`None`]
    /// if a problem occured.
    ///
    /// * `model_path` - the path to the model directory.
    #[must_use]
    pub fn new(model_path: impl Into<String>) -> Option<Self> {
        let model_path_c = CString::new(model_path.into()).ok()?;
        let model_ptr = unsafe { vosk_batch_model_new(model_path_c.as_ptr()) };

        Some(Self(NonNull::new(model_ptr)?))
    }

    /// Waits for inferencing to finish
    pub fn wait(&self) {
        unsafe { vosk_batch_model_wait(self.0.as_ptr()) };
    }
}

impl Drop for BatchModel {
    fn drop(&mut self) {
        unsafe { vosk_batch_model_free(self.0.as_ptr()) }
    }
}

unsafe impl Send for BatchModel {}
unsafe impl Sync for BatchModel {}