Commit
•
971a795
1
Parent(s):
2b09f60
Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def image_transform_grounding_for_vis(init_image):
|
|
62 |
image, _ = transform(init_image, None) # 3, h, w
|
63 |
return image
|
64 |
|
65 |
-
model = load_model_hf(
|
66 |
|
67 |
def segment(image, sam_model, boxes):
|
68 |
sam_model.set_image(image)
|
|
|
62 |
image, _ = transform(init_image, None) # 3, h, w
|
63 |
return image
|
64 |
|
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)
|