codezakh commited on
Commit
e7a55f5
1 Parent(s): 3c3e0f8

Fix "How to use" snippet

Browse files

The current "How To Use Snippet" cannot be copy-pasted and executed because of two minor problems:

L31: Trailing comma on the import that causes an error in python.
`device` is undefined

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-base"
 
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-base"
34
+ device = "cuda:0"
35
 
36
  processor = AutoProcessor.from_pretrained(model_id)
37
  model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)