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

chore: Refactor preprocess function to always remove background and resize foreground image

Browse files
Files changed (1) hide show
  1. app.py +7 -13
app.py CHANGED
@@ -55,12 +55,11 @@ def check_input_image(input_image):
55
  raise gr.Error("No image selected!")
56
 
57
 
58
- def preprocess(input_image, do_remove_background):
59
- rembg_session = rembg.new_session() if do_remove_background else None
60
 
61
- if do_remove_background:
62
- input_image = remove_background(input_image, rembg_session)
63
- input_image = resize_foreground(input_image, 0.85)
64
 
65
  return input_image
66
 
@@ -215,13 +214,9 @@ with gr.Blocks() as demo:
215
  type="pil",
216
  interactive=False
217
  )
218
- with gr.Row():
219
- with gr.Group():
220
- do_remove_background = gr.Checkbox(
221
- label="Remove Background", value=True)
222
  with gr.Row():
223
  submit_process = gr.Button(
224
- "Process Image", elem_id="process", variant="primary")
225
  with gr.Row(variant="panel"):
226
  gr.Examples(
227
  examples=[os.path.join("examples", img_name) for img_name in sorted(
@@ -273,11 +268,10 @@ 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, 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=[
 
55
  raise gr.Error("No image selected!")
56
 
57
 
58
+ def preprocess(input_image):
59
+ rembg_session = rembg.new_session()
60
 
61
+ input_image = remove_background(input_image, rembg_session)
62
+ input_image = resize_foreground(input_image, 0.85)
 
63
 
64
  return input_image
65
 
 
214
  type="pil",
215
  interactive=False
216
  )
 
 
 
 
217
  with gr.Row():
218
  submit_process = gr.Button(
219
+ "Remove Background", elem_id="process", variant="primary")
220
  with gr.Row(variant="panel"):
221
  gr.Examples(
222
  examples=[os.path.join("examples", img_name) for img_name in sorted(
 
268
  mv_images = gr.State()
269
 
270
  submit_prompt.click(fn=generate_image, inputs=[prompt], outputs=input_image).success(
271
+ fn=preprocess, inputs=[input_image], outputs=[processed_image]
272
  )
273
  submit_process.click(fn=check_input_image, inputs=[input_image]).success(
274
+ fn=preprocess, inputs=[input_image], outputs=[processed_image],
 
275
  )
276
  submit_mesh.click(fn=generate_mvs, inputs=[processed_image, sample_steps, sample_seed], outputs=[mv_images, mv_show_images]).success(
277
  fn=make3d, inputs=[mv_images], outputs=[