Spaces:

tyriaa
/

Segmentation_project

Runtime error

tyriaa commited on Nov 26, 2024

Commit

88b4248

1 Parent(s): a79e15b

Initialisation 000200

Files changed (1) hide show

app.py CHANGED Viewed

@@ -113,8 +113,8 @@ load_training_status()
 os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "0"
 # Initialize SAM Predictor
-MODEL_CFG = r"C:\codes\sam2\segment-anything-2\sam2\configs\sam2.1\sam2.1_hiera_l.yaml"
-CHECKPOINT = r"C:\codes\sam2\segment-anything-2\checkpoints\sam2.1_hiera_large.pt"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 predictor = Predictor(MODEL_CFG, CHECKPOINT, DEVICE)

 os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "0"
 # Initialize SAM Predictor
+MODEL_CFG = r"sam2/sam2_hiera_l.yaml"
+CHECKPOINT = r"sam2/checkpoints/sam2.1_hiera_large.pt"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 predictor = Predictor(MODEL_CFG, CHECKPOINT, DEVICE)