diff --git a/.envrc b/.envrc new file mode 100644 index 0000000..3550a30 --- /dev/null +++ b/.envrc @@ -0,0 +1 @@ +use flake diff --git a/.gitignore b/.gitignore index ea8c4bf..2d5df85 100644 --- a/.gitignore +++ b/.gitignore @@ -1 +1,2 @@ /target +.direnv diff --git a/src/model.rs b/src/model.rs index 43777fa..2f4d28d 100644 --- a/src/model.rs +++ b/src/model.rs @@ -17,7 +17,7 @@ impl ModelInfo { /// BiRefNet General Lite model pub const BIREFNET_LITE: ModelInfo = ModelInfo { name: "birefnet-general-lite", - url: "https://upload.wikimedia.org/wikipedia/commons/4/47/PNG_transparency_demonstration_1.png", + url: "https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-general-bb_swin_v1_tiny-epoch_232.onnx", sha256: None, // We'll skip verification for now input_size: (1024, 1024), };