michaelapplydesign commited on
Commit
f5b68ac
·
1 Parent(s): eff32bf
Files changed (1) hide show
  1. models.py +4 -2
models.py CHANGED
@@ -85,14 +85,16 @@ def make_inpainting(positive_prompt: str,
85
 
86
  flush()
87
  st.success(f"{pipe.queue_size} images in the queue, can take up to {(pipe.queue_size+1) * 10} seconds")
88
- generated_image = pipe(image=image,
89
  mask_image=mask_image,
90
  prompt=positive_prompt,
91
  negative_prompt=negative_prompt,
92
  num_inference_steps=50,
93
  height=HEIGHT,
94
  width=WIDTH,
95
- ).images[0]
 
 
96
  generated_image = postprocess_image_masking(generated_image, image, mask_image_postproc)
97
 
98
  return generated_image
 
85
 
86
  flush()
87
  st.success(f"{pipe.queue_size} images in the queue, can take up to {(pipe.queue_size+1) * 10} seconds")
88
+ resp = pipe(image=image,
89
  mask_image=mask_image,
90
  prompt=positive_prompt,
91
  negative_prompt=negative_prompt,
92
  num_inference_steps=50,
93
  height=HEIGHT,
94
  width=WIDTH,
95
+ )
96
+ print("RESP !!!!",resp)
97
+ generated_image = resp.images[0]
98
  generated_image = postprocess_image_masking(generated_image, image, mask_image_postproc)
99
 
100
  return generated_image