himipo commited on
Commit
6cf6df2
·
1 Parent(s): 4459426
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -178,7 +178,7 @@ def inference_pipeline(
178
  @spaces.GPU
179
  def gpu_inference(
180
  image: Image.Image,
181
- score_thresh: float = 0.8, # UIのデフォルト値と統一
182
  selected_classes: List[str] = None,
183
  ):
184
  """Spaces ZeroGPU が検出できるようにデコレータ付きの推論関数を用意"""
@@ -216,13 +216,13 @@ with gr.Blocks(title="DEIMv2 Floorplan Symbol Detection") as demo:
216
  score_thresh = gr.Slider(
217
  minimum=0.0,
218
  maximum=1.0,
219
- value=0.8,
220
  step=0.05,
221
  label="スコア閾値",
222
  )
223
  selected_classes = gr.CheckboxGroup(
224
  choices=class_names,
225
- value=class_names, # デフォルトで全クラスを選択
226
  label="検出するクラス",
227
  info="選択したクラスの検出結果のみが表示されます",
228
  )
 
178
  @spaces.GPU
179
  def gpu_inference(
180
  image: Image.Image,
181
+ score_thresh: float = 0.9, # UIのデフォルト値と統一
182
  selected_classes: List[str] = None,
183
  ):
184
  """Spaces ZeroGPU が検出できるようにデコレータ付きの推論関数を用意"""
 
216
  score_thresh = gr.Slider(
217
  minimum=0.0,
218
  maximum=1.0,
219
+ value=0.9,
220
  step=0.05,
221
  label="スコア閾値",
222
  )
223
  selected_classes = gr.CheckboxGroup(
224
  choices=class_names,
225
+ value=["door1"], # デフォルトでdoor1のみ選択
226
  label="検出するクラス",
227
  info="選択したクラスの検出結果のみが表示されます",
228
  )