djl234 commited on
Commit
c08de50
·
verified ·
1 Parent(s): eb9155a

Update app.py

Browse files

[feat] choose model

Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -224,7 +224,18 @@ with gr.Blocks(title="交互式图像组分割系统") as demo:
224
  mode = gr.Radio(["多图协同分割", "点提示交互分割","框提示交互分割"],
225
  value="多图协同分割",
226
  label="运行模式")
227
-
 
 
 
 
 
 
 
 
 
 
 
228
  # 使用Tab容器替代独立Blocks
229
  with gr.Tabs() as mode_container:
230
  with gr.Tab("多图模式", id=0) as tab1:
 
224
  mode = gr.Radio(["多图协同分割", "点提示交互分割","框提示交互分割"],
225
  value="多图协同分割",
226
  label="运行模式")
227
+ with gr.Row():
228
+ mode = gr.Radio(
229
+ ["多图协同分割", "点提示交互分割","框提示交互分割"],
230
+ value="多图协同分割",
231
+ label="运行模式"
232
+ )
233
+ model_selector = gr.Dropdown(
234
+ choices=["RepViT-SAM", "EdgeSAM", "SAM-H"],
235
+ value="SAM-H",
236
+ label="选择模型",
237
+ container=False # 去除默认容器边框
238
+ )
239
  # 使用Tab容器替代独立Blocks
240
  with gr.Tabs() as mode_container:
241
  with gr.Tab("多图模式", id=0) as tab1: