summaryrefslogtreecommitdiff
path: root/makima/src/server/state.rs
diff options
context:
space:
mode:
authorsoryu <soryu@soryu.co>2025-12-20 21:39:09 +0000
committersoryu <soryu@soryu.co>2025-12-23 14:47:18 +0000
commit84fee5ce2ae30fb2381c99b9b223b8235b962869 (patch)
tree68459d685a279a899571becbea054f2ea314bf9e /makima/src/server/state.rs
parent01088f4f1915e36a7d0d8d8756f62f8207a48911 (diff)
downloadsoryu-84fee5ce2ae30fb2381c99b9b223b8235b962869.tar.gz
soryu-84fee5ce2ae30fb2381c99b9b223b8235b962869.zip
Remove TTS endpoint using chatterbox
The library still remains, but this complicates deployment due to the large size of the model, so it is removed for now
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),
})
}
}