Update app.py

app.py

@@ -49,7 +49,7 @@ def resize_image(image):
 
 
 @spaces.GPU
-def generate_(prompt, negative_prompt, grayscale_image, num_steps, controlnet_conditioning_scale, seed):
+def generate_(prompt, negative_prompt, pose_image, num_steps, controlnet_conditioning_scale, seed):
     generator = torch.Generator("cuda").manual_seed(seed)    
     images = pipe(
     prompt, negative_prompt=negative_prompt, image=grayscale_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
@@ -67,7 +67,7 @@ def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditio
   
     images = generate_(prompt, negative_prompt, pose_image, num_steps, controlnet_conditioning_scale, seed)
 
-    return [grayscale_image,images[0]]
+    return [pose_image,images[0]]
     
 block = gr.Blocks().queue()