Hugging Face's logo

IDEA-Research
/
grounding-dino-tiny

Zero-Shot Object Detection
Transformers
PyTorch
Safetensors
grounding-dino
vision
Model card Files
xet
Community
6
nielsr HF Staff commited on
Commit
ec37772
·
verified ·
1 Parent(s): e95baf9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -28,9 +28,10 @@ import requests
28
 
29
  import torch
30
  from PIL import Image
31
- from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection,
32
 
33
  model_id = "IDEA-Research/grounding-dino-tiny"
 
34
 
35
  processor = AutoProcessor.from_pretrained(model_id)
36
  model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)
 
28
 
29
  import torch
30
  from PIL import Image
31
+ from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection
32
 
33
  model_id = "IDEA-Research/grounding-dino-tiny"
34
+ device = "cuda" if torch.cuda.is_available() else "cpu"
35
 
36
  processor = AutoProcessor.from_pretrained(model_id)
37
  model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)