Hugging Face's logo

Spaces:
mattmdjaga
/
segment_anything_base
Runtime error

App Files Community
1
mattmdjaga commited on
Commit
d399289
·
1 Parent(s): 390940a

Removed device='cpu'

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -8,7 +8,6 @@ import cv2
8
  from typing import List
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
- device = 'cpu'
12
 
13
  # Load model and processor
14
  model = SamModel.from_pretrained("facebook/sam-vit-base").to(device)
 
8
  from typing import List
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
11
 
12
  # Load model and processor
13
  model = SamModel.from_pretrained("facebook/sam-vit-base").to(device)