Spaces:

curt-park
/

segment-anything-with-clip

Runtime error

curt-park commited on Apr 11, 2023

Commit

b267f43

•

1 Parent(s): 8760721

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ def load_mask_generator() -> SamAutomaticMaskGenerator:
  checkpoint = os.path.join(CHECKPOINT_PATH, CHECKPOINT_NAME)
  if not os.path.exists(checkpoint):
  urllib.request.urlretrieve(CHECKPOINT_URL, checkpoint)
- sam = sam_model_registry[MODEL_TYPE](checkpoint=CHECKPOINT_PATH).to(device)
  mask_generator = SamAutomaticMaskGenerator(sam)
  return mask_generator

  checkpoint = os.path.join(CHECKPOINT_PATH, CHECKPOINT_NAME)
  if not os.path.exists(checkpoint):
  urllib.request.urlretrieve(CHECKPOINT_URL, checkpoint)
+ sam = sam_model_registry[MODEL_TYPE](checkpoint=checkpoint).to(device)
  mask_generator = SamAutomaticMaskGenerator(sam)
  return mask_generator