mdeville/remove_background
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

wip

Browse source
This commit is contained in:
Matthew Deville 2026-01-22 23:39:46 +01:00
parent 609afde727
commit 6e965c4a18
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/main.rs
View file

@ -20,19 +20,20 @@ struct Args {
#[arg(help = "URL of the image to download and process")] #[arg(help = "URL of the image to download and process")]
url: String, 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( #[arg(
short,
long, long,
group = "model_selection", group = "model_selection",
default_value = "bria", default_value = "bria",
help = "Model to use: 'bria' or 'birefnet-lite' (mutually exclusive with --model-path)" help = "Model to use: 'bria' or 'birefnet-lite' (mutually exclusive with --model-path)"
)] )]
model: String, 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] #[show_image::main]