Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -66,8 +66,8 @@ def sam_process(input_image):
|
|
66 |
image = Image.open(input_image)
|
67 |
image = np.array(image.convert("RGB"))
|
68 |
|
69 |
-
sam2_checkpoint = "./checkpoints/
|
70 |
-
model_cfg = "
|
71 |
|
72 |
sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cuda")
|
73 |
|
|
|
66 |
image = Image.open(input_image)
|
67 |
image = np.array(image.convert("RGB"))
|
68 |
|
69 |
+
sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
|
70 |
+
model_cfg = "./sam2_configs/sam2_hiera_t.yaml"
|
71 |
|
72 |
sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cuda")
|
73 |
|