Hugging Face's logo

Spaces:
abhishek
/
StableSAM
Runtime error

App Files Community
4
abhishek HF staff commited on
Commit
e1d0b29
•
1 Parent(s): 6b79de9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import colorsys
11
 
12
  sam_checkpoint = "sam_vit_h_4b8939.pth"
13
  model_type = "vit_h"
14
- device = "cuda"
15
 
16
 
17
  sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
 
11
 
12
  sam_checkpoint = "sam_vit_h_4b8939.pth"
13
  model_type = "vit_h"
14
+ device = "cpu"
15
 
16
 
17
  sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)