Update README.md

README.md

@@ -16,9 +16,9 @@ device = torch.device('cuda')
 sd3_vae = AutoencoderKL.from_pretrained("stabilityai/stable-diffusion-3.5-large", subfolder="vae")
 
 pipeline = DiffusionPipeline.from_pretrained(
-    "StonyBrook-CVLab/pixcell-256-diffusers",
+    "StonyBrook-CVLab/PixCell-256",
     vae=sd3_vae,
-    custom_pipeline="StonyBrook-CVLab/pixcell-pipeline",
+    custom_pipeline="StonyBrook-CVLab/PixCell-pipeline",
     trust_remote_code=True,
 )
 
@@ -66,7 +66,7 @@ from PIL import Image
 from huggingface_hub import hf_hub_download
 
 # This is an example image we provide
-path = hf_hub_download(repo_id="StonyBrook-CVLab/pixcell-256-diffusers", filename="test_image.jpg")
+path = hf_hub_download(repo_id="StonyBrook-CVLab/PixCell-256", filename="test_image.jpg")
 image = Image.open(path)
 
 # Extract UNI from random patches in the image