anvilarth commited on
Commit
9cc52c4
1 Parent(s): ebc0141

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -170,10 +170,9 @@ class GradioWindow():
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
 
 
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=GROUNDING_DINO_CHECKPOINT_PATH,
176
  device=self.device
177
  )
178