Update README.md (#6)
Browse files- Update README.md (f8222717b1892bb3e548d4f07538c24e39c596f7)
Co-authored-by: Lain <[email protected]>
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
|