ZhengPeng7 not-lain commited on
Commit
99ec976
1 Parent(s): c9bfa66

Update README.md (#6)

Browse files

- Update README.md (f8222717b1892bb3e548d4f07538c24e39c596f7)


Co-authored-by: Lain <not-lain@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -64,7 +64,7 @@ pip install -qr https://raw.githubusercontent.com/ZhengPeng7/BiRefNet/main/requi
64
  ```python
65
  # Load BiRefNet with weights
66
  from transformers import AutoModelForImageSegmentation
67
- birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet', trust_remote_code=True)
68
  ```
69
 
70
  #### Use codes from GitHub + weights from HuggingFace
@@ -81,7 +81,7 @@ cd BiRefNet
81
  from models.birefnet import BiRefNet
82
 
83
  # Load weights from Hugging Face Models
84
- birefnet = BiRefNet.from_pretrained('zhengpeng7/BiRefNet')
85
  ```
86
 
87
  #### Use codes from GitHub + weights from HuggingFace
 
64
  ```python
65
  # Load BiRefNet with weights
66
  from transformers import AutoModelForImageSegmentation
67
+ birefnet = AutoModelForImageSegmentation.from_pretrained('ZhengPeng7/BiRefNet', trust_remote_code=True)
68
  ```
69
 
70
  #### Use codes from GitHub + weights from HuggingFace
 
81
  from models.birefnet import BiRefNet
82
 
83
  # Load weights from Hugging Face Models
84
+ birefnet = BiRefNet.from_pretrained('ZhengPeng7/BiRefNet')
85
  ```
86
 
87
  #### Use codes from GitHub + weights from HuggingFace