Ashoka74 commited on
Commit
2e90656
ยท
verified ยท
1 Parent(s): 5b3477f

Update merged_app2.py

Browse files
Files changed (1) hide show
  1. merged_app2.py +2 -1
merged_app2.py CHANGED
@@ -1658,6 +1658,7 @@ with gr.Blocks() as app:
1658
  orientation_result.select(use_orientation, inputs=None, outputs=extracted_fg)
1659
 
1660
  dummy_image_for_outputs = gr.Image(visible=False, label='Result', type='pil')
 
1661
 
1662
 
1663
  with gr.Column():
@@ -1798,7 +1799,7 @@ with gr.Blocks() as app:
1798
  # outputs=[result_gallery, output_bg],
1799
  # run_on_click=True, examples_per_page=1024
1800
  # )
1801
- augment_prompt.click(generate_description, inputs=[prompt,None], outputs=[prompt])
1802
  ips = [extracted_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source]
1803
 
1804
  relight_button.click(fn=process_relight, inputs=ips, outputs=[result_gallery]).then(clear_memory, inputs=[], outputs=[])
 
1658
  orientation_result.select(use_orientation, inputs=None, outputs=extracted_fg)
1659
 
1660
  dummy_image_for_outputs = gr.Image(visible=False, label='Result', type='pil')
1661
+ dummy_image_for_prompt_augmentation = gr.Image(type="pil", label="Dummy image", height=480)
1662
 
1663
 
1664
  with gr.Column():
 
1799
  # outputs=[result_gallery, output_bg],
1800
  # run_on_click=True, examples_per_page=1024
1801
  # )
1802
+ augment_prompt.click(generate_description, inputs=[prompt, dummy_image_for_prompt_augmentation], outputs=[prompt])
1803
  ips = [extracted_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source]
1804
 
1805
  relight_button.click(fn=process_relight, inputs=ips, outputs=[result_gallery]).then(clear_memory, inputs=[], outputs=[])