c4181/whisper-worker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix model parameter

Browse source
This commit is contained in:
Christopher Moyer 2024-12-17 18:06:30 -05:00
parent b34f97fa4d
commit 30d7a061b6
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/main.rs
View file

@ -16,7 +16,7 @@ use log::{info, warn};
#[tokio::main] #[tokio::main]
async fn main() { async fn main() {
let ascii = r#" let ascii = r#"
__ ___ _ _ __ __ __ _ __ ___ _ _ __ __ __ _
\ \ / / |_ (_) ___ | '_ \ ___ _ _ o O O\ \ / / ___ _ _ | |__ ___ _ _ \ \ / / |_ (_) ___ | '_ \ ___ _ _ o O O\ \ / / ___ _ _ | |__ ___ _ _
\ \/\/ /| ' \ | | (_-< | .__/ / -_) | '_| o \ \/\/ / / _ \ | '_| | / / / -_) | '_| \ \/\/ /| ' \ | | (_-< | .__/ / -_) | '_| o \ \/\/ / / _ \ | '_| | / / / -_) | '_|
\_/\_/ |_||_| _|_|_ /__/_ |_|__ \___| _|_|_ TS__[O] \_/\_/ \___/ _|_|_ |_\_\ \___| _|_|_ \_/\_/ |_||_| _|_|_ /__/_ |_|__ \___| _|_|_ TS__[O] \_/\_/ \___/ _|_|_ |_\_\ \___| _|_|_
@ -88,7 +88,7 @@ fn transcribe_call(file_path: &Path) {
let output_directory = file_path.parent().unwrap(); let output_directory = file_path.parent().unwrap();
let output = Command::new("whisperx") let output = Command::new("whisperx")
.args(["--language", "en"]) .args(["--language", "en"])
.args(["--model", "largev3"]) .args(["--model", "large-v3"])
.args(["--batch_size", "4"]) .args(["--batch_size", "4"])
.args(["--compute_type", "int8"]) .args(["--compute_type", "int8"])
.args(["--output_format", "txt"]) .args(["--output_format", "txt"])