浏览代码

fix model url

Daniel Gatis 9 月之前
父节点
当前提交
1101c1521c
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      rembg/sessions/bria_rmbg.py

+ 1 - 1
rembg/sessions/bria_rmbg.py

@@ -60,7 +60,7 @@ class BriaRmBgSession(BaseSession):
         """
         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)