Ashoka74 commited on
Commit
5104ffd
Β·
verified Β·
1 Parent(s): afe48d5

Update gradio_demo.py

Browse files
Files changed (1) hide show
  1. gradio_demo.py +11 -6
gradio_demo.py CHANGED
@@ -1004,13 +1004,18 @@ with block:
1004
  # run_on_click=True, examples_per_page=1024
1005
  # )
1006
  ips = [input_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source]
1007
- relight_button.click(fn=process_relight, inputs=ips, outputs=[extracted_fg, result_gallery])
1008
  example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
1009
  example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
 
 
 
 
 
1010
  find_objects_button.click(
1011
  fn=process_image,
1012
  inputs=[input_fg, text_prompt],
1013
- outputs=[extracted_objects, extracted_fg]
1014
  )
1015
 
1016
  with gr.Tab("Background", visible=False):
@@ -1140,19 +1145,19 @@ with block:
1140
  extract_button.click(
1141
  fn=extract_foreground,
1142
  inputs=[input_image],
1143
- outputs=[extracted_fg, x_slider, y_slider]
1144
  )
1145
 
1146
  find_objects_button.click(
1147
  fn=process_image,
1148
  inputs=[input_image, text_prompt],
1149
- outputs=[extracted_objects, extracted_fg, x_slider, y_slider]
1150
  )
1151
 
1152
  get_depth_button.click(
1153
  fn=get_depth,
1154
  inputs=[input_bg],
1155
- outputs=[depth_image]
1156
  )
1157
 
1158
  # def update_position(background, x_pos, y_pos, scale):
@@ -1294,7 +1299,7 @@ with block:
1294
  relight_button_bg.click(
1295
  fn=process_relight_with_position,
1296
  inputs=ips_bg,
1297
- outputs=[result_gallery]
1298
  )
1299
 
1300
 
 
1004
  # run_on_click=True, examples_per_page=1024
1005
  # )
1006
  ips = [input_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source]
1007
+ relight_button.click(fn=process_relight, inputs=ips, outputs=[extracted_fg, result_gallery], show_progress=True)
1008
  example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
1009
  example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
1010
+ extract_button.click(
1011
+ fn=extract_foreground,
1012
+ inputs=[input_fg],
1013
+ outputs=[extracted_fg, x_slider, y_slider], show_progress=True
1014
+ )
1015
  find_objects_button.click(
1016
  fn=process_image,
1017
  inputs=[input_fg, text_prompt],
1018
+ outputs=[extracted_objects, extracted_fg], show_progress=True
1019
  )
1020
 
1021
  with gr.Tab("Background", visible=False):
 
1145
  extract_button.click(
1146
  fn=extract_foreground,
1147
  inputs=[input_image],
1148
+ outputs=[extracted_fg, x_slider, y_slider], show_progress=True
1149
  )
1150
 
1151
  find_objects_button.click(
1152
  fn=process_image,
1153
  inputs=[input_image, text_prompt],
1154
+ outputs=[extracted_objects, extracted_fg, x_slider, y_slider], show_progress=True
1155
  )
1156
 
1157
  get_depth_button.click(
1158
  fn=get_depth,
1159
  inputs=[input_bg],
1160
+ outputs=[depth_image], show_progress=True
1161
  )
1162
 
1163
  # def update_position(background, x_pos, y_pos, scale):
 
1299
  relight_button_bg.click(
1300
  fn=process_relight_with_position,
1301
  inputs=ips_bg,
1302
+ outputs=[result_gallery], show_progress=True
1303
  )
1304
 
1305