Update README.md

README.md

@@ -2,7 +2,7 @@
 license: apache-2.0
 ---
 
-# Model Card for Segment Anything Model (SAM) - ViT Big (ViT-B) version
+# Model Card for Segment Anything Model (SAM) - ViT Base (ViT-B) version
 
 <p>
 	<img src="https://s3.amazonaws.com/moonup/production/uploads/62441d1d9fdefb55a0b7d12c/F1LWM9MXjHJsiAtgBFpDP.png" alt="Model architecture">
@@ -50,8 +50,8 @@ from PIL import Image
 import requests
 from transformers import SamModel, SamProcessor
 
-model = SamModel.from_pretrained("facebook/sam-vit-big")
-processsor = SamProcessor.from_pretrained("facebook/sam-vit-big")
+model = SamModel.from_pretrained("facebook/sam-vit-base")
+processsor = SamProcessor.from_pretrained("facebook/sam-vit-base")
 
 img_url = "https://huggingface.co/ybelkada/segment-anything/resolve/main/assets/car.png"
 raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")