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
37
38
39
40
41
42
43
44
45
46
//! Speech recognition engines for transcription.
//!
//! This module contains implementations of different speech recognition engines
//! that can be used for audio transcription. Each engine has its own requirements
//! for model formats and provides different capabilities.
//!
//! # Available Engines
//!
//! ## Whisper Engine
//!
//! OpenAI's Whisper model implementation:
//! - **Model Format**: Single GGML format file (`.bin`)
//! - **Models**: tiny, base, small, medium, large variants
//! - **Features**: Multi-language support, robust performance
//! - **Example**: `whisper-medium-q4_1.bin`
//!
//! ## Parakeet Engine
//!
//! NVIDIA NeMo Parakeet model implementation:
//! - **Model Format**: Directory containing model files
//! - **Features**: Flexible timestamp granularity, quantization support
//! - **Performance**: Optimized for speed with Int8 quantization
//! - **Example**: `parakeet-v0.3/` directory
//!
//! # Usage Comparison
//!
//! ```rust,no_run
//! use transcribe_rs::{TranscriptionEngine, engines::{whisper::WhisperEngine, parakeet::ParakeetEngine}};
//! use std::path::PathBuf;
//!
//! // Whisper: Single file model
//! let mut whisper = WhisperEngine::new();
//! whisper.load_model(&PathBuf::from("models/whisper-medium-q4_1.bin"))?;
//!
//! // Parakeet: Directory model
//! let mut parakeet = ParakeetEngine::new();
//! parakeet.load_model(&PathBuf::from("models/parakeet-v0.3"))?;
//!
//! // Both engines implement the same TranscriptionEngine trait
//! let whisper_result = whisper.transcribe_file(&PathBuf::from("audio.wav"), None)?;
//! let parakeet_result = parakeet.transcribe_file(&PathBuf::from("audio.wav"), None)?;
//! # Ok::<(), Box<dyn std::error::Error>>(())
//! ```