ZhengPeng7
commited on
Commit
·
a7e29d3
1
Parent(s):
6f22264
Ensemble BiRefNet codes to HF to load model with weights in one line.
Browse files
README.md
CHANGED
@@ -35,7 +35,7 @@ from models.birefnet import BiRefNet
|
|
35 |
birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
|
36 |
```
|
37 |
|
38 |
-
#### Use codes +
|
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
|
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
|