nielsr HF staff commited on
Commit
b98cc67
1 Parent(s): bc7d330

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ import torch
27
  from datasets import load_dataset
28
  from transformers import SegGptImageProcessor, SegGptForImageSegmentation
29
 
30
- model_id = "EduardoPacheco/seggpt-vit-large"
31
  image_processor = SegGptImageProcessor.from_pretrained(checkpoint)
32
  model = SegGptForImageSegmentation.from_pretrained(checkpoint)
33
 
 
27
  from datasets import load_dataset
28
  from transformers import SegGptImageProcessor, SegGptForImageSegmentation
29
 
30
+ model_id = "BAAI/seggpt-vit-large"
31
  image_processor = SegGptImageProcessor.from_pretrained(checkpoint)
32
  model = SegGptForImageSegmentation.from_pretrained(checkpoint)
33