Hugging Face's logo

Spaces:
dragonSwing
/
annotate-anything
Sleeping

App Files Community
dragonSwing commited on
Commit
f1213ed
·
1 Parent(s): f695925

Fix grounding dino mismatch device

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ sam_predictor = SamPredictor(sam)
80
  sam_automask_generator = SamAutomaticMaskGenerator(sam)
81
 
82
  grounding_dino_model = DinoModel(
83
- model_config_path=dino_config_file, model_checkpoint_path=dino_checkpoint
84
  )
85
 
86
 
 
80
  sam_automask_generator = SamAutomaticMaskGenerator(sam)
81
 
82
  grounding_dino_model = DinoModel(
83
+ model_config_path=dino_config_file, model_checkpoint_path=dino_checkpoint, device=device
84
  )
85
 
86