Spaces:

Roboflow
/

SoM

Runtime error

SkalskiP commited on Nov 20, 2023

Commit

1400af8

1 Parent(s): 0c52132

Fix `SAM_CHECKPOINT` path

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,8 @@ from segment_anything import sam_model_registry, SamAutomaticMaskGenerator
 HOME = os.getenv("HOME")
 DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
-SAM_CHECKPOINT = os.path.join(HOME, "weights/sam_vit_h_4b8939.pth")
 SAM_MODEL_TYPE = "vit_h"
 MARKDOWN = """

 HOME = os.getenv("HOME")
 DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
+SAM_CHECKPOINT = os.path.join(HOME, "app/weights/sam_vit_h_4b8939.pth")
+# SAM_CHECKPOINT = "weights/sam_vit_h_4b8939.pth"
 SAM_MODEL_TYPE = "vit_h"
 MARKDOWN = """