Update app.py

app.py

@@ -85,12 +85,12 @@ def resize_image_old(image):
 
 
 @spaces.GPU
-def generate_(prompt, negative_prompt, pose_image, num_steps, controlnet_conditioning_scale, seed):
+def generate_(prompt, negative_prompt, pose_image, input_image, num_steps, controlnet_conditioning_scale, seed):
     generator = torch.Generator("cuda").manual_seed(seed)    
     images = pipe(
     prompt, negative_prompt=negative_prompt, image=pose_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
-    generator=generator, height=pose_image.size[1], width=pose_image.size[0],
-    ).images
+    generator=generator, height=input_image.size[1], width=input_image.size[0],
+    ).images 
     return images
 
 @spaces.GPU
@@ -101,7 +101,7 @@ def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditio
     
     pose_image = openpose(input_image, include_body=True, include_hand=True, include_face=True)
   
-    images = generate_(prompt, negative_prompt, pose_image, num_steps, controlnet_conditioning_scale, seed)
+    images = generate_(prompt, negative_prompt, pose_image, input_image, num_steps, controlnet_conditioning_scale, seed)
 
     return [pose_image,images[0]]