not-lain commited on
Commit
0b8a0c4
·
verified ·
1 Parent(s): 37e94a6

Update README.md

Browse files

updating readme after removing the weights leaving briaai the sole owner of the weights

Files changed (1) hide show
  1. README.md +2 -2
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("not-lain/CustomCodeForRMBG",trust_remote_code=True)
11
  ```
12
  or load the pipeline
13
  ```python
14
  from transformers import pipeline
15
 
16
- pipe = pipeline("image-segmentation", model="not-lain/CustomCodeForRMBG", trust_remote_code=True)
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