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>
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('
|
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('
|
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
|