Update README.md
Browse files
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 = "
|
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 |
|