Fix `SAM_CHECKPOINT` path
Browse files
app.py
CHANGED
|
@@ -12,7 +12,8 @@ from segment_anything import sam_model_registry, SamAutomaticMaskGenerator
|
|
| 12 |
HOME = os.getenv("HOME")
|
| 13 |
DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 14 |
|
| 15 |
-
SAM_CHECKPOINT = os.path.join(HOME, "weights/sam_vit_h_4b8939.pth")
|
|
|
|
| 16 |
SAM_MODEL_TYPE = "vit_h"
|
| 17 |
|
| 18 |
MARKDOWN = """
|
|
|
|
| 12 |
HOME = os.getenv("HOME")
|
| 13 |
DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 14 |
|
| 15 |
+
SAM_CHECKPOINT = os.path.join(HOME, "app/weights/sam_vit_h_4b8939.pth")
|
| 16 |
+
# SAM_CHECKPOINT = "weights/sam_vit_h_4b8939.pth"
|
| 17 |
SAM_MODEL_TYPE = "vit_h"
|
| 18 |
|
| 19 |
MARKDOWN = """
|