Update README.md

README.md

@@ -54,8 +54,8 @@ from PIL import Image
 import requests
 from transformers import SamHQModel, SamHQProcessor
 
-model = SamHQModel.from_pretrained("sushmanth/sam_hq_vit_b")
-processor = SamHQProcessor.from_pretrained("sushmanth/sam_hq_vit_b")
+model = SamHQModel.from_pretrained("syscv-community/sam-hq-vit-base")
+processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-base")
 
 img_url = "https://raw.githubusercontent.com/SysCV/sam-hq/refs/heads/main/demo/input_imgs/example1.png"
 raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")
@@ -79,7 +79,7 @@ The pipeline is made for automatic mask generation. The following snippet demons
 
 ```python
 from transformers import pipeline
-generator = pipeline("mask-generation", model="sushmanth/sam_hq_vit_b", device=0, points_per_batch=256)
+generator = pipeline("mask-generation", model="syscv-community/sam-hq-vit-base", device=0, points_per_batch=256)
 image_url = "https://raw.githubusercontent.com/SysCV/sam-hq/refs/heads/main/demo/input_imgs/example1.png"
 outputs = generator(image_url, points_per_batch=256)
 ```
@@ -210,8 +210,8 @@ import torch
 from transformers import SamHQModel, SamHQProcessor
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model = SamHQModel.from_pretrained("sushmanth/sam_hq_vit_b").to(device)
-processor = SamHQProcessor.from_pretrained("sushmanth/sam_hq_vit_b")
+model = SamHQModel.from_pretrained("syscv-community/sam-hq-vit-base").to(device)
+processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-base")
 
 from PIL import Image
 import requests