segment_anything_webui

Sleeping

AIBoy1993 commited on Apr 8, 2023

Commit

81a1f3d

1 Parent(s): e7cc8df

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,7 +60,7 @@ with gr.Blocks() as demo:
             # 选择模型类型
             model_type = gr.Dropdown(["vit_b", "vit_l", "vit_h"], value='vit_b', label="选择模型")
             # 选择device
-            device = gr.Dropdown(["cpu", "cuda"], value='cuda', label="选择你的硬件")
     # 参数
     with gr.Accordion(label='参数调整', open=False):

             # 选择模型类型
             model_type = gr.Dropdown(["vit_b", "vit_l", "vit_h"], value='vit_b', label="选择模型")
             # 选择device
+            device = gr.Dropdown(["cpu"], value='cpu', label="选择你的硬件")
     # 参数
     with gr.Accordion(label='参数调整', open=False):