Spaces:

tyriaa
/

project

Runtime error

tyriaa commited on Nov 26, 2024

Commit

d617907

1 Parent(s): 7d98737

Initial commit153

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"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)

 os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "0"
 # Initialize SAM Predictor
+MODEL_CFG = r"https://huggingface.co/spaces/tyriaa/project/blob/main/sam2/sam2_hiera_l.yaml"
+CHECKPOINT = r"https://huggingface.co/spaces/tyriaa/project/blob/main/sam2/sam2.1_hiera_large.pt"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 predictor = Predictor(MODEL_CFG, CHECKPOINT, DEVICE)