summaryrefslogtreecommitdiff
path: root/makima/src/server/state.rs
diff options
context:
space:
mode:
Diffstat (limited to 'makima/src/server/state.rs')
-rw-r--r--makima/src/server/state.rs7
1 files changed, 0 insertions, 7 deletions
diff --git a/makima/src/server/state.rs b/makima/src/server/state.rs
index 8eaf788..c38359d 100644
--- a/makima/src/server/state.rs
+++ b/makima/src/server/state.rs
@@ -4,7 +4,6 @@ use std::sync::Arc;
use tokio::sync::Mutex;
use crate::listen::{DiarizationConfig, ParakeetTDT, Sortformer};
-use crate::tts::ChatterboxTTS;
/// Shared application state containing ML models.
///
@@ -14,8 +13,6 @@ pub struct AppState {
pub parakeet: Mutex<ParakeetTDT>,
/// Speaker diarization model (Sortformer)
pub sortformer: Mutex<Sortformer>,
- /// Text-to-speech model (ChatterboxTTS)
- pub chatterbox: Mutex<ChatterboxTTS>,
}
impl AppState {
@@ -24,11 +21,9 @@ impl AppState {
/// # Arguments
/// * `parakeet_model_dir` - Path to the Parakeet STT model directory
/// * `sortformer_model_path` - Path to the Sortformer diarization model file
- /// * `tts_model_dir` - Optional path to the ChatterboxTTS model directory
pub fn new(
parakeet_model_dir: &str,
sortformer_model_path: &str,
- tts_model_dir: Option<&str>,
) -> Result<Self, Box<dyn std::error::Error + Send + Sync>> {
let parakeet = ParakeetTDT::from_pretrained(parakeet_model_dir, None)?;
let sortformer = Sortformer::with_config(
@@ -36,12 +31,10 @@ impl AppState {
None,
DiarizationConfig::callhome(),
)?;
- let chatterbox = ChatterboxTTS::from_pretrained(tts_model_dir)?;
Ok(Self {
parakeet: Mutex::new(parakeet),
sortformer: Mutex::new(sortformer),
- chatterbox: Mutex::new(chatterbox),
})
}
}