Spaces:
Running
Running
Commit
·
306c0aa
1
Parent(s):
87e31cc
Maybe fix preload paths
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def show_box(box, ax):
|
|
62 |
|
63 |
|
64 |
|
65 |
-
sam_checkpoint = r"~/.cache/huggingface/hub/sam_vit_l_0b3195.pth"
|
66 |
model_type = "vit_l"
|
67 |
|
68 |
sam = sam_model_registry[model_type](checkpoint=sam_checkpoint, device=device)
|
@@ -180,7 +180,7 @@ def predict_one(net, image, mask, box, transforms, hypar, device):
|
|
180 |
|
181 |
hypar = {} # paramters for inferencing
|
182 |
|
183 |
-
hypar["model_path"] ="~/.cache/huggingface/hub"
|
184 |
hypar["restore_model"] = "DIS-SAM-checkpoint.pth"
|
185 |
hypar["model_digit"] = "full"
|
186 |
hypar["input_size"] = [1024, 1024]
|
|
|
62 |
|
63 |
|
64 |
|
65 |
+
sam_checkpoint = r"~/.cache/huggingface/hub/andzhang01/segment_anything/sam_vit_l_0b3195.pth"
|
66 |
model_type = "vit_l"
|
67 |
|
68 |
sam = sam_model_registry[model_type](checkpoint=sam_checkpoint, device=device)
|
|
|
180 |
|
181 |
hypar = {} # paramters for inferencing
|
182 |
|
183 |
+
hypar["model_path"] ="~/.cache/huggingface/hub/jwlarocque/DIS-SAM"
|
184 |
hypar["restore_model"] = "DIS-SAM-checkpoint.pth"
|
185 |
hypar["model_digit"] = "full"
|
186 |
hypar["input_size"] = [1024, 1024]
|