abreza commited on
Commit
b20e0b4
·
1 Parent(s): 7a04400

fix: Update Gradio UI layout for image generation and processing

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -273,7 +273,11 @@ with gr.Blocks() as demo:
273
  mv_images = gr.State()
274
 
275
  submit_prompt.click(fn=generate_image, inputs=[prompt], outputs=input_image).success(
276
- fn=preprocess, inputs=[input_image, True], outputs=[processed_image]
 
 
 
 
277
  )
278
  submit_mesh.click(fn=generate_mvs, inputs=[processed_image, sample_steps, sample_seed], outputs=[mv_images, mv_show_images]).success(
279
  fn=make3d, inputs=[mv_images], outputs=[
 
273
  mv_images = gr.State()
274
 
275
  submit_prompt.click(fn=generate_image, inputs=[prompt], outputs=input_image).success(
276
+ fn=preprocess, inputs=[input_image, do_remove_background], outputs=[processed_image]
277
+ )
278
+ submit_process.click(fn=check_input_image, inputs=[input_image]).success(
279
+ fn=preprocess, inputs=[input_image,
280
+ do_remove_background], outputs=[processed_image],
281
  )
282
  submit_mesh.click(fn=generate_mvs, inputs=[processed_image, sample_steps, sample_seed], outputs=[mv_images, mv_show_images]).success(
283
  fn=make3d, inputs=[mv_images], outputs=[