BAAI
/

seggpt-vit-large

Model card Files Files and versions

nielsr HF Staff commited on Feb 22, 2024

Commit

b98cc67

·

verified ·

1 Parent(s): bc7d330

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ import torch
 from datasets import load_dataset
 from transformers import SegGptImageProcessor, SegGptForImageSegmentation
-model_id = "EduardoPacheco/seggpt-vit-large"
 image_processor = SegGptImageProcessor.from_pretrained(checkpoint)
 model = SegGptForImageSegmentation.from_pretrained(checkpoint)

 from datasets import load_dataset
 from transformers import SegGptImageProcessor, SegGptForImageSegmentation
+model_id = "BAAI/seggpt-vit-large"
 image_processor = SegGptImageProcessor.from_pretrained(checkpoint)
 model = SegGptForImageSegmentation.from_pretrained(checkpoint)