polejowska commited on
Commit
4b6f2a1
1 Parent(s): acc27e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -71,7 +71,7 @@ def detect_objects(model_name, image_input, threshold, display_mask=False, img_i
71
 
72
 
73
  def set_example_image(example: list):
74
- return gr.Image.update(value=example[0])
75
 
76
 
77
  with gr.Blocks(css=css) as app:
@@ -144,7 +144,8 @@ with gr.Blocks(css=css) as app:
144
  queue=True,
145
  )
146
  example_images.click(
147
- fn=set_example_image, inputs=[example_images], outputs=[img_input]
148
  )
149
 
150
  app.launch(enable_queue=True)
 
 
71
 
72
 
73
  def set_example_image(example: list):
74
+ return gr.Image.update(value=example[0]), gr.Image.update(value=example[1])
75
 
76
 
77
  with gr.Blocks(css=css) as app:
 
144
  queue=True,
145
  )
146
  example_images.click(
147
+ fn=set_example_image, inputs=[example_images], outputs=[img_input, img_input_mask]
148
  )
149
 
150
  app.launch(enable_queue=True)
151
+