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