Hugging Face's logo

Spaces:
martintomov
/
InsectModelZoo
Running

App Files Community
1
Martin Tomov commited on
Commit
81b2e04
·
verified ·
1 Parent(s): cfccf84

IDEA-Research/grounding-dino-base

Browse files
Files changed (1) hide show
  1. gsl_utils.py +1 -1
gsl_utils.py CHANGED
@@ -12,7 +12,7 @@ from huggingface_hub import hf_hub_download
12
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
13
 
14
  def load_groundingdino_model(device='cpu'):
15
- model = pipeline(model="ShilongLiu/GroundingDINO", task="zero-shot-object-detection", device=device)
16
  return model
17
 
18
  def load_sam_model(device='cpu'):
 
12
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
13
 
14
  def load_groundingdino_model(device='cpu'):
15
+ model = pipeline(model="IDEA-Research/grounding-dino-base", task="zero-shot-object-detection", device=device)
16
  return model
17
 
18
  def load_sam_model(device='cpu'):