ZhengPeng7 commited on
Commit
a7bfa64
1 Parent(s): 17f5dfc

Capitalize the BiRefNet when used as the HF model name.

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- library_name: birefnet
3
  tags:
4
  - background-removal
5
  - mask-generation
@@ -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>
@@ -58,7 +58,7 @@ Visit our GitHub repo: [https://github.com/ZhengPeng7/BiRefNet](https://github.c
58
  ```python
59
  # Load BiRefNet with weights
60
  from transformers import AutoModelForImageSegmentation
61
- birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/birefnet', trust_remote_code=True)
62
  ```
63
 
64
  #### Use codes from GitHub + weights from HuggingFace
@@ -75,7 +75,7 @@ cd BiRefNet
75
  from models.birefnet import BiRefNet
76
 
77
  # Load weights from Hugging Face Models
78
- birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
79
  ```
80
 
81
  #### Use codes from GitHub + weights from HuggingFace
 
1
  ---
2
+ library_name: BiRefNet
3
  tags:
4
  - background-removal
5
  - mask-generation
 
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>
 
58
  ```python
59
  # Load BiRefNet with weights
60
  from transformers import AutoModelForImageSegmentation
61
+ birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet', trust_remote_code=True)
62
  ```
63
 
64
  #### Use codes from GitHub + weights from HuggingFace
 
75
  from models.birefnet import BiRefNet
76
 
77
  # Load weights from Hugging Face Models
78
+ birefnet = BiRefNet.from_pretrained('zhengpeng7/BiRefNet')
79
  ```
80
 
81
  #### Use codes from GitHub + weights from HuggingFace