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

fix clap args so cli works

Browse Source
This commit is contained in:
Bel LaPointe
2023-12-19 22:17:37 -05:00
parent 95f638d3d0
commit 0a0827a7b2
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,8 +9,9 @@ use std::io::Write;
#[derive(Parser, Debug)] #[derive(Parser, Debug)]
pub struct Flags { pub struct Flags {
#[arg(long, default_value = "./models/ggml-tiny.en.bin")] #[arg(long, default_value = "../models/ggml-tiny.en.bin")]
pub model_path: Option<String>, pub model_path: Option<String>,
#[arg(long, default_value = None)]
pub model_buffer: Option<Vec<u8>>, pub model_buffer: Option<Vec<u8>>,
#[arg(long, default_value = "8")] #[arg(long, default_value = "8")]
@@ -25,6 +26,7 @@ pub struct Flags {
#[arg(long, default_value = "0.3")] #[arg(long, default_value = "0.3")]
pub stream_tail: f32, pub stream_tail: f32,
#[arg(long, default_value = None)]
pub wav: Option<String>, pub wav: Option<String>,
#[arg(long, default_value = "false")] #[arg(long, default_value = "false")]