fffiloni commited on
Commit
6eee3ae
1 Parent(s): a88740c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/sam2_hiera_large.pt"
70
- model_cfg = "sam2_hiera_l.yaml"
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