IDEA-Research
/

grounding-dino-base

Zero-Shot Object Detection

Model card Files Files and versions Community

nielsr HF staff commited on Apr 26

Commit

e76a695

•

1 Parent(s): 33b593c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -31,7 +31,7 @@ from PIL import Image
 from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection
 model_id = "IDEA-Research/grounding-dino-base"
-device = "cuda:0"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)

 from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection
 model_id = "IDEA-Research/grounding-dino-base"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)