anvilarth commited on
Commit
ebc0141
·
verified ·
1 Parent(s): 09e25af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -170,9 +170,10 @@ class GradioWindow():
170
  self.sam.to(device=self.device)
171
  self.sam_predictor = SamPredictor(self.sam)
172
 
 
173
  self.grounding_dino_model = Model(
174
  model_config_path=self.GROUNDING_DINO_CONFIG_PATH,
175
- model_checkpoint_path=self.GROUNDING_DINO_CHECKPOINT_PATH,
176
  device=self.device
177
  )
178
 
 
170
  self.sam.to(device=self.device)
171
  self.sam_predictor = SamPredictor(self.sam)
172
 
173
+ dino_model = torch.utils.model_zoo.load_url(GROUNDING_DINO_CHECKPOINT_PATH)
174
  self.grounding_dino_model = Model(
175
  model_config_path=self.GROUNDING_DINO_CONFIG_PATH,
176
+ model_checkpoint_path=dino_model,
177
  device=self.device
178
  )
179