From 1101c1521c3c7f04b9fa68a4f0d61b338f5ee9e7 Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Tue, 3 Dec 2024 17:22:49 -0300 Subject: [PATCH] fix model url --- rembg/sessions/bria_rmbg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rembg/sessions/bria_rmbg.py b/rembg/sessions/bria_rmbg.py index 990af33a..a1bda828 100644 --- a/rembg/sessions/bria_rmbg.py +++ b/rembg/sessions/bria_rmbg.py @@ -60,7 +60,7 @@ def download_models(cls, *args, **kwargs): """ fname = f"{cls.name(*args, **kwargs)}.onnx" pooch.retrieve( - "https://huggingface.co/briaai/RMBG-2.0/resolve/main/onnx/model.onnx", + "https://github.com/danielgatis/rembg/releases/download/v0.0.0/bria-rmbg-2.0.onnx", ( None if cls.checksum_disabled(*args, **kwargs)