Spaces:

martintomov
/

InsectModelZoo

Runtime error

Martin Tomov commited on Jun 22, 2024

Commit

81b2e04

verified ·

1 Parent(s): cfccf84

IDEA-Research/grounding-dino-base

Files changed (1) hide show

gsl_utils.py CHANGED Viewed

@@ -12,7 +12,7 @@ from huggingface_hub import hf_hub_download
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 def load_groundingdino_model(device='cpu'):
-    model = pipeline(model="ShilongLiu/GroundingDINO", task="zero-shot-object-detection", device=device)
     return model
 def load_sam_model(device='cpu'):

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 def load_groundingdino_model(device='cpu'):
+    model = pipeline(model="IDEA-Research/grounding-dino-base", task="zero-shot-object-detection", device=device)
     return model
 def load_sam_model(device='cpu'):