jwlarocque commited on
Commit
306c0aa
·
1 Parent(s): 87e31cc

Maybe fix preload paths

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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]