xqt commited on
Commit
e35d910
·
verified ·
1 Parent(s): ce02f1d

UPD: Set default to CPU

Browse files
Files changed (1) hide show
  1. SegmentAnything2AssistApp.py +1 -1
SegmentAnything2AssistApp.py CHANGED
@@ -409,7 +409,7 @@ with gradio.Blocks() as base_app:
409
  )
410
  with gradio.Column():
411
  base_gpu_choice = gradio.Dropdown(
412
- ["cpu", "cuda"], value="cuda", label="Device Choice"
413
  )
414
  base_model_choice.change(
415
  change_base_model, inputs=[base_model_choice, base_gpu_choice]
 
409
  )
410
  with gradio.Column():
411
  base_gpu_choice = gradio.Dropdown(
412
+ ["cpu", "cuda"], value="cpu", label="Device Choice"
413
  )
414
  base_model_choice.change(
415
  change_base_model, inputs=[base_model_choice, base_gpu_choice]