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