ZhengPeng7 commited on
Commit
a7e29d3
·
1 Parent(s): 6f22264

Ensemble BiRefNet codes to HF to load model with weights in one line.

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -35,7 +35,7 @@ from models.birefnet import BiRefNet
35
  birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
36
  ```
37
 
38
- #### Use codes + Weights from HuggingFace
39
  > Only use the weights on HuggingFace -- Pro: No need to download BiRefNet codes manually; Con: Codes on HuggingFace might not be latest version (I'll try to keep them always latest).
40
 
41
  ```python
@@ -44,7 +44,7 @@ from transformers import AutoModelForImageSegmentation
44
  birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/birefnet', trust_remote_code=True)
45
  ```
46
 
47
- #### Use the Loaded BiRefNet for inference
48
  ```python
49
  # Imports
50
  from PIL import Image
 
35
  birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
36
  ```
37
 
38
+ #### Use codes + weights from HuggingFace
39
  > Only use the weights on HuggingFace -- Pro: No need to download BiRefNet codes manually; Con: Codes on HuggingFace might not be latest version (I'll try to keep them always latest).
40
 
41
  ```python
 
44
  birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/birefnet', trust_remote_code=True)
45
  ```
46
 
47
+ #### Use the loaded BiRefNet for inference
48
  ```python
49
  # Imports
50
  from PIL import Image