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

chore: Update Gradio UI layout for image generation and processing

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -198,7 +198,6 @@ with gr.Blocks() as demo:
198
  label='Enter your prompt (English)', scale=2)
199
  submit_prompt = gr.Button(
200
  'Generate Image', scale=1, variant='primary')
201
- img = gr.Image()
202
 
203
  with gr.Row(variant="panel"):
204
  with gr.Column():
@@ -273,10 +272,8 @@ with gr.Blocks() as demo:
273
 
274
  mv_images = gr.State()
275
 
276
- submit_prompt.click(fn=generate_image, inputs=[prompt], outputs=img)
277
- submit_process.click(fn=check_input_image, inputs=[input_image]).success(
278
- fn=preprocess, inputs=[input_image,
279
- do_remove_background], outputs=[processed_image],
280
  )
281
  submit_mesh.click(fn=generate_mvs, inputs=[processed_image, sample_steps, sample_seed], outputs=[mv_images, mv_show_images]).success(
282
  fn=make3d, inputs=[mv_images], outputs=[
 
198
  label='Enter your prompt (English)', scale=2)
199
  submit_prompt = gr.Button(
200
  'Generate Image', scale=1, variant='primary')
 
201
 
202
  with gr.Row(variant="panel"):
203
  with gr.Column():
 
272
 
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=[