BhumikaMak commited on
Commit
8d393b3
·
verified ·
1 Parent(s): 1a33b04
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -159,7 +159,6 @@ body {
159
 
160
  }
161
 
162
-
163
  """
164
 
165
  import netron
@@ -309,15 +308,12 @@ with gr.Blocks(css=custom_css) as demo:
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
 
320
-
321
  with gr.Column():
322
  sample_display = gr.Image(
323
  value=load_sample_image(default_sample),
 
159
 
160
  }
161
 
 
162
  """
163
 
164
  import netron
 
308
  )
309
 
310
  selected_models = gr.CheckboxGroup(
311
+ choices=["yolov5", "yolov8s"], # Only the models that can be selected
312
  value=["yolov5"],
313
  label="Select Model(s)",
314
  )
 
 
315
  run_button = gr.Button("Run", elem_id="run-button")
316
 
 
317
  with gr.Column():
318
  sample_display = gr.Image(
319
  value=load_sample_image(default_sample),