BhumikaMak commited on
Commit
1a33b04
·
verified ·
1 Parent(s): f86d70c

updated: select models

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -309,10 +309,11 @@ with gr.Blocks(css=custom_css) as demo:
309
  )
310
 
311
  selected_models = gr.CheckboxGroup(
312
- choices=["yolov5", "yolov8s", "Bring Your Own Model (Coming Soon)"],
313
  value=["yolov5"],
314
  label="Select Model(s)",
315
  )
 
316
  #with gr.Row(elem_classes="custom-row"):
317
  run_button = gr.Button("Run", elem_id="run-button")
318
 
 
309
  )
310
 
311
  selected_models = gr.CheckboxGroup(
312
+ choices=["yolov5", "yolov8s"],
313
  value=["yolov5"],
314
  label="Select Model(s)",
315
  )
316
+ selected_models.update(choices=["yolov5", "yolov8s", "Bring Your Own Model (Coming Soon)"])
317
  #with gr.Row(elem_classes="custom-row"):
318
  run_button = gr.Button("Run", elem_id="run-button")
319