wip
This commit is contained in:
parent
609afde727
commit
6e965c4a18
1 changed files with 7 additions and 6 deletions
13
src/main.rs
13
src/main.rs
|
|
@ -20,19 +20,20 @@ struct Args {
|
|||
#[arg(help = "URL of the image to download and process")]
|
||||
url: String,
|
||||
|
||||
#[arg(long, group = "model_selection", help = "Path to custom ONNX model")]
|
||||
model_path: Option<String>,
|
||||
|
||||
#[arg(long, help = "Skip model download, fail if not cached")]
|
||||
offline: bool,
|
||||
|
||||
#[arg(
|
||||
short,
|
||||
long,
|
||||
group = "model_selection",
|
||||
default_value = "bria",
|
||||
help = "Model to use: 'bria' or 'birefnet-lite' (mutually exclusive with --model-path)"
|
||||
)]
|
||||
model: String,
|
||||
|
||||
#[arg(long, group = "model_selection", help = "Path to custom ONNX model")]
|
||||
model_path: Option<String>,
|
||||
|
||||
#[arg(long, help = "Skip model download, fail if not cached")]
|
||||
offline: bool,
|
||||
}
|
||||
|
||||
#[show_image::main]
|
||||
|
|
|
|||
Loading…
Reference in a new issue