Spaces:
Runtime error
Runtime error
Commit ·
33402bc
1
Parent(s): 971a795
Update app.py
Browse files
app.py
CHANGED
|
@@ -65,7 +65,6 @@ def image_transform_grounding_for_vis(init_image):
|
|
| 65 |
model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
|
| 66 |
|
| 67 |
def segment(image, sam_model, boxes):
|
| 68 |
-
sam_model.set_image(image)
|
| 69 |
H, W, _ = image.shape
|
| 70 |
boxes_xyxy = box_ops.box_cxcywh_to_xyxy(boxes) * torch.Tensor([W, H, W, H])
|
| 71 |
|
|
|
|
| 65 |
model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
|
| 66 |
|
| 67 |
def segment(image, sam_model, boxes):
|
|
|
|
| 68 |
H, W, _ = image.shape
|
| 69 |
boxes_xyxy = box_ops.box_cxcywh_to_xyxy(boxes) * torch.Tensor([W, H, W, H])
|
| 70 |
|