BhumikaMak commited on
Commit
26e2298
·
1 Parent(s): 02640d8

Fix: model dependency + default sample

Browse files
Files changed (1) hide show
  1. app.py +9 -14
app.py CHANGED
@@ -37,11 +37,12 @@ with gr.Blocks() as interface:
37
  gr.Markdown("Upload an image or select a sample image to visualize object detection.")
38
 
39
  with gr.Row():
40
- uploaded_image = gr.Image(type="pil", label="Upload an Image", value = None)
41
  sample_selection = gr.Dropdown(
42
  choices=list(sample_images.keys()),
43
  label="Select a Sample Image",
44
- type="Sample 1",
 
45
  )
46
  sample_display = gr.Image(label="Sample Image Preview", value=default_sample_image)
47
  sample_selection.change(
@@ -57,17 +58,11 @@ with gr.Blocks() as interface:
57
  )
58
 
59
  result_gallery = gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500)
60
- if uploaded_image is not None:
61
- gr.Button("Run").click(
62
- fn=process_image,
63
- inputs=[uploaded_image, selected_models],
64
- outputs=result_gallery,
65
- )
66
- else:
67
- gr.Button("Run").click(
68
- fn=process_image,
69
- inputs=[sample_display, selected_models],
70
- outputs=result_gallery,
71
- )
72
 
73
  interface.launch()
 
37
  gr.Markdown("Upload an image or select a sample image to visualize object detection.")
38
 
39
  with gr.Row():
40
+ uploaded_image = gr.Image(type="pil", label="Upload an Image")
41
  sample_selection = gr.Dropdown(
42
  choices=list(sample_images.keys()),
43
  label="Select a Sample Image",
44
+ value="Sample 1", # Default selection
45
+ type="value", # Correct type parameter
46
  )
47
  sample_display = gr.Image(label="Sample Image Preview", value=default_sample_image)
48
  sample_selection.change(
 
58
  )
59
 
60
  result_gallery = gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500)
61
+
62
+ gr.Button("Run").click(
63
+ fn=process_image,
64
+ inputs=[uploaded_image, selected_models, sample_selection],
65
+ outputs=result_gallery,
66
+ )
 
 
 
 
 
 
67
 
68
  interface.launch()