diff options
| author | soryu <soryu@soryu.co> | 2025-12-21 00:40:04 +0000 |
|---|---|---|
| committer | soryu <soryu@soryu.co> | 2025-12-23 14:47:18 +0000 |
| commit | 55cacf6e1a087c0fa6950a1ddeb09060f787e541 (patch) | |
| tree | 0b8e754eb16c829fc0ee7c8f4ba66fe75b4f3ebf /makima/src/bin/server.rs | |
| parent | 84fee5ce2ae30fb2381c99b9b223b8235b962869 (diff) | |
| download | soryu-55cacf6e1a087c0fa6950a1ddeb09060f787e541.tar.gz soryu-55cacf6e1a087c0fa6950a1ddeb09060f787e541.zip | |
Add EOU detection and streaming diarization
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))?, ); |