Update app.py

app.py

@@ -33,8 +33,8 @@ def generate_(prompt, negative_prompt, control_tensor, num_steps, controlnet_con
         image = control_tensor,
         generator=generator
     ).images[0]
-    result_image = paste_fg_over_image(gen_img, image, mask)
-    return result_image
+    
+    return gen_img
 
 @spaces.GPU
 def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditioning_scale, seed):
@@ -43,9 +43,10 @@ def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditio
     mask = remove_bg_from_image(image)
     control_tensor = get_control_image_tensor(pipe.vae, image, mask)    
   
-    images = generate_(prompt, negative_prompt, control_tensor, num_steps, controlnet_conditioning_scale, seed)
+    gen_image = generate_(prompt, negative_prompt, control_tensor, num_steps, controlnet_conditioning_scale, seed)
+    result_image = paste_fg_over_image(gen_image, image, mask)
 
-    return [depth_image, images[0]]
+    return [mask ,result_image]