BhumikaMak commited on
Commit
66370da
·
1 Parent(s): f7b8e0e

Fix: model dependency + default sample

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -26,6 +26,8 @@ sample_images = {
26
  "Sample 1": os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
27
  "Sample 2": os.path.join(os.getcwd(), "data/xai/sample2.jpg"),
28
  }
 
 
29
  def load_sample_image(sample_name):
30
  if sample_name in sample_images:
31
  try:
@@ -46,11 +48,11 @@ with gr.Blocks() as interface:
46
  label="Select a Sample Image",
47
  type="value",
48
  )
49
- sample_display = gr.Image(label="Sample Image Preview", value=None)
50
  sample_selection.change(fn=load_sample_image, inputs=sample_selection, outputs=sample_display)
51
 
52
  selected_models = gr.CheckboxGroup(
53
- choices=["yolov3", "yolov8s"],
54
  value=["yolov5"], # Default model
55
  label="Select Model(s)",
56
  )
 
26
  "Sample 1": os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
27
  "Sample 2": os.path.join(os.getcwd(), "data/xai/sample2.jpg"),
28
  }
29
+ default_sample_image = load_sample_image("Sample 1")
30
+
31
  def load_sample_image(sample_name):
32
  if sample_name in sample_images:
33
  try:
 
48
  label="Select a Sample Image",
49
  type="value",
50
  )
51
+ sample_display = gr.Image(label="Sample Image Preview", value=default_sample_image)
52
  sample_selection.change(fn=load_sample_image, inputs=sample_selection, outputs=sample_display)
53
 
54
  selected_models = gr.CheckboxGroup(
55
+ choices=["yolov5", "yolov8s"],
56
  value=["yolov5"], # Default model
57
  label="Select Model(s)",
58
  )