IDEA-Research
/

grounding-dino-tiny

Zero-Shot Object Detection

Model card Files Files and versions Community

nielsr HF staff commited on Apr 26

Commit

ec37772

•

1 Parent(s): e95baf9

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -28,9 +28,10 @@ import requests
 import torch
 from PIL import Image
-from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection,
 model_id = "IDEA-Research/grounding-dino-tiny"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)

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