Spaces:
Build error
Build error
UPD: Set default to CPU
Browse files
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="
|
| 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]
|