Spaces:

MarcHabib
/

segmentation_yolo_sam

Running

MarcHabib commited on Nov 26, 2024

Commit

a547007

1 Parent(s): 09e3ceb

Updated paths

Files changed (1) hide show

app.py CHANGED Viewed

@@ -86,8 +86,8 @@ load_training_status()
 os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "0"
 # Initialize SAM Predictor
-MODEL_CFG = r"configs\sam2.1\sam2.1_hiera_l.yaml"
-CHECKPOINT = r"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 = "configs/sam2.1/sam2.1_hiera_l.yaml"
+CHECKPOINT = "checkpoints/sam2.1_hiera_large.pt"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 predictor = Predictor(MODEL_CFG, CHECKPOINT, DEVICE)