Update README.md
Browse filesupdating readme after removing the weights leaving briaai the sole owner of the weights
README.md
CHANGED
|
@@ -7,13 +7,13 @@ pipeline_tag: image-segmentation
|
|
| 7 |
either load the model
|
| 8 |
```python
|
| 9 |
from transformers import AutoModelForImageSegmentation
|
| 10 |
-
model = AutoModelForImageSegmentation.from_pretrained("
|
| 11 |
```
|
| 12 |
or load the pipeline
|
| 13 |
```python
|
| 14 |
from transformers import pipeline
|
| 15 |
|
| 16 |
-
pipe = pipeline("image-segmentation", model="
|
| 17 |
|
| 18 |
numpy_mask = pipe("img_path") # outputs numpy mask
|
| 19 |
|
|
|
|
| 7 |
either load the model
|
| 8 |
```python
|
| 9 |
from transformers import AutoModelForImageSegmentation
|
| 10 |
+
model = AutoModelForImageSegmentation.from_pretrained("briaai/RMBG-1.4",revision ="refs/pr/9",trust_remote_code=True)
|
| 11 |
```
|
| 12 |
or load the pipeline
|
| 13 |
```python
|
| 14 |
from transformers import pipeline
|
| 15 |
|
| 16 |
+
pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4",revision ="refs/pr/9", trust_remote_code=True)
|
| 17 |
|
| 18 |
numpy_mask = pipe("img_path") # outputs numpy mask
|
| 19 |
|