iSEE-Laboratory
/

llmdet_base

Zero-Shot Object Detection

mm-grounding-dino

Model card Files Files and versions

rziga commited on Jul 26

Commit

3537869

·

verified ·

1 Parent(s): a6b1114

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ from transformers.image_utils import load_image
 # Prepare processor and model
-model_id = "rziga/llmdet_base"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)

 # Prepare processor and model
+model_id = "iSEE-Laboratory/llmdet_base"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)