Hugging Face's logo

Spaces:
spdraptor
/
Photo_edit
Paused

App Files Community
spdraptor commited on
Commit
ed51f1c
·
1 Parent(s): e003fb9

fix/try sam

Browse files
Files changed (2) hide show
  1. modules/__pycache__/masking_module.cpython-310.pyc +0 -0
  2. modules/masking_module.py +1 -1
modules/__pycache__/masking_module.cpython-310.pyc CHANGED
Binary files a/modules/__pycache__/masking_module.cpython-310.pyc and b/modules/__pycache__/masking_module.cpython-310.pyc differ
 
modules/masking_module.py CHANGED
@@ -31,7 +31,7 @@ def load_sam_image_model(
31
  config: str = SAM_CONFIG,
32
  checkpoint: str = SAM_CHECKPOINT
33
  ) -> SAM2ImagePredictor:
34
- model = build_sam2(config, checkpoint, device=device)
35
  return SAM2ImagePredictor(sam_model=model)
36
 
37
 
 
31
  config: str = SAM_CONFIG,
32
  checkpoint: str = SAM_CHECKPOINT
33
  ) -> SAM2ImagePredictor:
34
+ model = build_sam2(config, checkpoint)
35
  return SAM2ImagePredictor(sam_model=model)
36
 
37