AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-18 02:25:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix model url

Browse Source
This commit is contained in:
Daniel Gatis 2024-12-03 17:22:49 -03:00
parent 692ef5c7fb
commit 1101c1521c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rembg/sessions/bria_rmbg.py
View File

@ -60,7 +60,7 @@ class BriaRmBgSession(BaseSession):
""" """
fname = f"{cls.name(*args, **kwargs)}.onnx" fname = f"{cls.name(*args, **kwargs)}.onnx"
pooch.retrieve( 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 None
if cls.checksum_disabled(*args, **kwargs) if cls.checksum_disabled(*args, **kwargs)