facebook
/

sam-vit-huge

Mask Generation

Inference Endpoints

Model card Files Files and versions Community

ybelkada commited on Apr 19, 2023

Commit

f956ef0

·

1 Parent(s): f0d3fe6

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -48,8 +48,8 @@ The SAM model is made up of 3 modules:
 ```python
 >>> from PIL import Image
 >>> import requests
->>> from transformers import SamModelForMaskedGeneration, SamProcessor
->>> model = SamModelForMaskedGeneration.from_pretrained("facebook/sam-vit-huge")
 >>> processsor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
 >>> img_url = "https://huggingface.co/ybelkada/segment-anything/resolve/main/assets/car.png"

 ```python
 >>> from PIL import Image
 >>> import requests
+>>> from transformers import SamModel, SamProcessor
+>>> model = SamModel.from_pretrained("facebook/sam-vit-huge")
 >>> processsor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
 >>> img_url = "https://huggingface.co/ybelkada/segment-anything/resolve/main/assets/car.png"