ZhengPeng7 commited on
Commit
16e8da8
1 Parent(s): 29b773a

Update README for portrait-related content.

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -34,7 +34,7 @@ pipeline_tag: image-segmentation
34
  <a href='https://drive.google.com/drive/folders/1s2Xe0cjq-2ctnJBR24563yMSCOu4CcxM'><img src='https://img.shields.io/badge/Drive-Stuff-green'></a>&ensp;
35
  <a href='LICENSE'><img src='https://img.shields.io/badge/License-MIT-yellow'></a>&ensp;
36
  <a href='https://huggingface.co/spaces/ZhengPeng7/BiRefNet_demo'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20HF%20Spaces-BiRefNet-blue'></a>&ensp;
37
- <a href='https://huggingface.co/ZhengPeng7/birefnet'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20HF%20Models-BiRefNet-blue'></a>&ensp;
38
  <a href='https://colab.research.google.com/drive/14Dqg7oeBkFEtchaHLNpig2BcdkZEogba?usp=drive_link'><img src='https://img.shields.io/badge/Single_Image_Inference-F9AB00?style=for-the-badge&logo=googlecolab&color=525252'></a>&ensp;
39
  <a href='https://colab.research.google.com/drive/1MaEiBfJ4xIaZZn0DqKrhydHB8X97hNXl#scrollTo=DJ4meUYjia6S'><img src='https://img.shields.io/badge/Inference_&_Evaluation-F9AB00?style=for-the-badge&logo=googlecolab&color=525252'></a>&ensp;
40
  </div>
@@ -64,7 +64,7 @@ cd BiRefNet
64
  ```python
65
  # Load weights
66
  from models.birefnet import BiRefNet
67
- birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
68
  ```
69
 
70
  #### Use codes + weights from HuggingFace
@@ -73,7 +73,7 @@ birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
73
  ```python
74
  # Load BiRefNet with weights
75
  from transformers import AutoModelForImageSegmentation
76
- birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/birefnet', trust_remote_code=True)
77
  ```
78
 
79
  #### Use the loaded BiRefNet for inference
 
34
  <a href='https://drive.google.com/drive/folders/1s2Xe0cjq-2ctnJBR24563yMSCOu4CcxM'><img src='https://img.shields.io/badge/Drive-Stuff-green'></a>&ensp;
35
  <a href='LICENSE'><img src='https://img.shields.io/badge/License-MIT-yellow'></a>&ensp;
36
  <a href='https://huggingface.co/spaces/ZhengPeng7/BiRefNet_demo'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20HF%20Spaces-BiRefNet-blue'></a>&ensp;
37
+ <a href='https://huggingface.co/ZhengPeng7/birefnet-portrait'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20HF%20Models-BiRefNet-blue'></a>&ensp;
38
  <a href='https://colab.research.google.com/drive/14Dqg7oeBkFEtchaHLNpig2BcdkZEogba?usp=drive_link'><img src='https://img.shields.io/badge/Single_Image_Inference-F9AB00?style=for-the-badge&logo=googlecolab&color=525252'></a>&ensp;
39
  <a href='https://colab.research.google.com/drive/1MaEiBfJ4xIaZZn0DqKrhydHB8X97hNXl#scrollTo=DJ4meUYjia6S'><img src='https://img.shields.io/badge/Inference_&_Evaluation-F9AB00?style=for-the-badge&logo=googlecolab&color=525252'></a>&ensp;
40
  </div>
 
64
  ```python
65
  # Load weights
66
  from models.birefnet import BiRefNet
67
+ birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet-portrait')
68
  ```
69
 
70
  #### Use codes + weights from HuggingFace
 
73
  ```python
74
  # Load BiRefNet with weights
75
  from transformers import AutoModelForImageSegmentation
76
+ birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/birefnet-portrait', trust_remote_code=True)
77
  ```
78
 
79
  #### Use the loaded BiRefNet for inference