diff --git a/src/main.rs b/src/main.rs index 3d18854..1ddd536 100644 --- a/src/main.rs +++ b/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, - - #[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, + + #[arg(long, help = "Skip model download, fail if not cached")] + offline: bool, } #[show_image::main]