From 8dfd74f84361ba644e3429e8db1ce3306eab3687 Mon Sep 17 00:00:00 2001
From: zmwv823 <913720128@qq.com>
Date: Tue, 3 Sep 2024 21:24:46 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=A8=A1=E5=9E=8B=EF=BC=8C?=
 =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=A8=A1=E5=9E=8B=E8=AF=B4=E6=98=8E=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md | 14 ++++++++++++++
 nodes.py  |  2 ++
 2 files changed, 16 insertions(+)

diff --git a/README.md b/README.md
index 757e7f1..f59da52 100644
--- a/README.md
+++ b/README.md
@@ -50,6 +50,20 @@ ___
 
 Thanks to BiRefNet repo owner [ZhengPeng7/BiRefNet](https://github.com/zhengpeng7/birefnet)
 
+```
+library_name: birefnet
+tags:
+  - background-removal
+  - mask-generation
+  - Dichotomous Image Segmentation:二分图像分割(DIS)，指分割成前景与背景，二个集合，分割出高精度效果。
+  - Camouflaged Object Detection:伪装物体检测(COD)，偏工程向，旨在识别“无缝”嵌入到周围环境中的物体，例如野生动物保护、军事侦察或者工业自动化。
+  - Salient Object Detection:显著性目标检测(SOD)，自动检测图像中最具视觉吸引力的部分。
+  - pytorch_model_hub_mixin
+  - model_hub_mixin
+repo_url: https://github.com/ZhengPeng7/BiRefNet
+pipeline_tag: image-segmentation
+license: mit
+```
 部分代码参考了 [ZHO-ZHO-ZHO/ComfyUI-BiRefNet-ZHO](https://github.com/ZHO-ZHO-ZHO/ComfyUI-BiRefNet-ZHO) 感谢！
 
 Some of the code references [ZHO-ZHO-ZHO/ComfyUI-BiRefNet-ZHO](https://github.com/ZHO-ZHO-ZHO/ComfyUI-BiRefNet-ZHO) Thanks!
diff --git a/nodes.py b/nodes.py
index e49ad23..e2a8b37 100644
--- a/nodes.py
+++ b/nodes.py
@@ -102,6 +102,8 @@ class BiRefNet_Hugo:
         rembg_list.insert(0, "Auto_DownLoad-ZhengPeng7/BiRefNet")
         rembg_list.insert(1, "Auto_DownLoad-ZhengPeng7/BiRefNet-DIS5K-TR_TEs")
         rembg_list.insert(2, "Auto_DownLoad-ZhengPeng7/BiRefNet-COD")
+        rembg_list.insert(3, "Auto_DownLoad-ZhengPeng7/BiRefNet-HRSOD")
+        rembg_list.insert(4, "Auto_DownLoad-ZhengPeng7/BiRefNet-portrait")
         return {
             "required": {
                 "model": (rembg_list, ),