diff options
Diffstat (limited to 'makima/src/bin/server.rs')
| -rw-r--r-- | makima/src/bin/server.rs | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/makima/src/bin/server.rs b/makima/src/bin/server.rs index 1964cae..06b6585 100644 --- a/makima/src/bin/server.rs +++ b/makima/src/bin/server.rs @@ -10,6 +10,7 @@ use makima::server::{run_server, state::AppState}; /// Default model paths relative to the working directory. const PARAKEET_MODEL_DIR: &str = "models/parakeet-tdt-0.6b-v3"; +const PARAKEET_EOU_DIR: &str = "models/realtime_eou_120m-v1-onnx"; const SORTFORMER_MODEL_PATH: &str = "models/diarization/diar_streaming_sortformer_4spk-v2.onnx"; #[tokio::main] @@ -28,7 +29,7 @@ async fn main() -> anyhow::Result<()> { // Load ML models let state = Arc::new( - AppState::new(PARAKEET_MODEL_DIR, SORTFORMER_MODEL_PATH) + AppState::new(PARAKEET_MODEL_DIR, PARAKEET_EOU_DIR, SORTFORMER_MODEL_PATH) .map_err(|e| anyhow::anyhow!("Failed to load models: {}", e))?, ); |