diff --git a/src-tauri/src/app_state.rs b/src-tauri/src/app_state.rs index e78b1ac..74d08cc 100644 --- a/src-tauri/src/app_state.rs +++ b/src-tauri/src/app_state.rs @@ -48,8 +48,7 @@ impl Default for ManagedAppState { impl ManagedAppState { pub fn sherpa_model_dir(&self) -> PathBuf { - self.model_dir - .join("sherpa-onnx-nemo-parakeet-tdt-0.6b-v3-int8") + crate::models::sherpa_model_dir(&self.model_dir) } pub fn snapshot(&self) -> Result { @@ -89,7 +88,12 @@ fn app_data_dir() -> PathBuf { #[cfg(target_os = "linux")] { std::env::var("HOME") - .map(|home| PathBuf::from(home).join(".local").join("share").join("transcript")) + .map(|home| { + PathBuf::from(home) + .join(".local") + .join("share") + .join("transcript") + }) .unwrap_or_else(|_| std::env::temp_dir().join("transcript")) } diff --git a/src-tauri/src/models/mod.rs b/src-tauri/src/models/mod.rs index 9e5adac..adba74b 100644 --- a/src-tauri/src/models/mod.rs +++ b/src-tauri/src/models/mod.rs @@ -8,7 +8,7 @@ use crate::error::{AppError, AppResult}; /// Modelo padrão usado pelo `SherpaOnnxTranscriber` (Etapa 2). /// É um pacote multi-arquivo do release `asr-models` do k2-fsa/sherpa-onnx. -pub const SHERPA_MODEL_ID: &str = "sherpa-onnx-nemo-parakeet-tdt-0.6b-v3-int8"; +pub const SHERPA_MODEL_ID: &str = "sherpa-onnx-nemo-parakeet-tdt-0.6b-v2-int8"; /// ID legado do modelo GGUF (mantido para o botão "Baixar modelo" não /// quebrar builds existentes que ainda apontam para o catálogo antigo). @@ -76,14 +76,14 @@ pub fn catalog() -> Vec { vec![ OfflineModel { id: SHERPA_MODEL_ID.to_string(), - display_name: "Parakeet TDT 0.6B v3 (sherpa-onnx int8)".to_string(), + display_name: "Parakeet TDT 0.6B v2 (sherpa-onnx int8)".to_string(), repository: "k2-fsa/sherpa-onnx asr-models".to_string(), quantization: "int8".to_string(), format: ModelFormat::Archive { - archive_url: "https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/sherpa-onnx-nemo-parakeet-tdt-0.6b-v3-int8.tar.bz2" + archive_url: "https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/sherpa-onnx-nemo-parakeet-tdt-0.6b-v2-int8.tar.bz2" .to_string(), - archive_size_bytes: 670_000_000, - extracted_subdir: "sherpa-onnx-nemo-parakeet-tdt-0.6b-v3-int8".to_string(), + archive_size_bytes: 640_000_000, + extracted_subdir: "sherpa-onnx-nemo-parakeet-tdt-0.6b-v2-int8".to_string(), required_files: vec![ "encoder.int8.onnx".to_string(), "decoder.int8.onnx".to_string(),