Hugging Face's logo

Spaces:
xqt
/
Segment-Anything-2-Assist
Running on Zero

App Files Community
1
xqt commited on
Commit
2e4869a
·
1 Parent(s): 6a53904

FIX: uses cuda by default

Browse files
Files changed (1) hide show
  1. SegmentAnything2AssistApp.py +1 -1
SegmentAnything2AssistApp.py CHANGED
@@ -50,7 +50,7 @@ DEBUG = False
50
 
51
 
52
  segment_anything2assist = SegmentAnything2Assist.SegmentAnything2Assist(
53
- model_name="sam2_hiera_tiny", device=torch.device("cpu")
54
  )
55
 
56
 
 
50
 
51
 
52
  segment_anything2assist = SegmentAnything2Assist.SegmentAnything2Assist(
53
+ model_name="sam2_hiera_tiny", device=torch.device("cuda")
54
  )
55
 
56