iSEE-Laboratory
/

llmdet_tiny

Zero-Shot Object Detection

mm-grounding-dino

Model card Files Files and versions Community

rziga commited on 14 days ago

Commit

d051991

·

verified ·

1 Parent(s): c2f9d4b

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_tiny"
 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_tiny"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)