chaoxu commited on
Commit
5fc36c4
1 Parent(s): 83bd11e

normal pred minor bug fix

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -0
gradio_app.py CHANGED
@@ -140,6 +140,7 @@ def gen_multiview(pipeline, pipeline_normal, predictor, input_image, scale_slide
140
  prompt='', guidance_scale=1, num_inference_steps=50, width=640, height=960
141
  ).images[0]
142
  subimages_normal = [image_normal.crop((x, y, x + side_len, y+side_len)) for y in range(0, image_normal.height, side_len) for x in range(0, image_normal.width, side_len)]
 
143
 
144
  if "Background Removal" in output_processing:
145
  out_images = []
 
140
  prompt='', guidance_scale=1, num_inference_steps=50, width=640, height=960
141
  ).images[0]
142
  subimages_normal = [image_normal.crop((x, y, x + side_len, y+side_len)) for y in range(0, image_normal.height, side_len) for x in range(0, image_normal.width, side_len)]
143
+ out_images_normal = subimages_normal
144
 
145
  if "Background Removal" in output_processing:
146
  out_images = []