Update app.py
Browse files
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=
|
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 |
|