gogs
/
rust-whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

minor refactor

master
Bel LaPointe 2023-12-20 14:55:33 -05:00
parent 589c6e65b8
commit f97f746ab9
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
rust-whisper-lib/src/lib.rs
View File

@ -199,21 +199,14 @@ struct Engine {
} }
fn new_engine(model_path: Option<String>, model_buffer: Option<Vec<u8>>, threads: i32) -> Result<Engine, String> { fn new_engine(model_path: Option<String>, model_buffer: Option<Vec<u8>>, threads: i32) -> Result<Engine, String> {
if model_path.is_some() { let whisper_context_result = match model_path {
let model_path = model_path.unwrap(); Some(model_path) => WhisperContext::new(&model_path),
return match WhisperContext::new(&model_path.clone()) { None => WhisperContext::new_from_buffer(&model_buffer.unwrap()),
Ok(ctx) => Ok(Engine{ctx: ctx, threads: threads}),
Err(msg) => Err(format!("failed to load {}: {}", model_path, msg)),
}; };
} match whisper_context_result {
if model_buffer.is_some() {
let model_buffer = model_buffer.unwrap();
return match WhisperContext::new_from_buffer(&model_buffer) {
Ok(ctx) => Ok(Engine{ctx: ctx, threads: threads}), Ok(ctx) => Ok(Engine{ctx: ctx, threads: threads}),
Err(msg) => Err(format!("failed to load buffer: {}", msg)), Err(msg) => Err(format!("failed to load model: {}", msg)),
};
} }
Err("neither model path nor buffer provided".to_string())
} }
impl Engine { impl Engine {