Update app.py

app.py

@@ -44,10 +44,10 @@ def resize_image(image):
 
 
 @spaces.GPU
-def generate_(prompt, negative_prompt, canny_image, num_steps, controlnet_conditioning_scale, seed):
+def generate_(prompt, negative_prompt, grayscale_image, num_steps, controlnet_conditioning_scale, seed):
     generator = torch.Generator("cuda").manual_seed(seed)    
     images = pipe(
-    prompt, negative_prompt=negative_prompt, image=canny_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
+    prompt, negative_prompt=negative_prompt, image=grayscale_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
     generator=generator,
     ).images
     return images
@@ -58,11 +58,11 @@ def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditio
     # resize input_image to 1024x1024
     input_image = resize_image(input_image)
     
-    canny_image = get_canny_filter(input_image)
+    grayscale_image = input_image.convert('L').convert('RGB')
   
-    images = generate_(prompt, negative_prompt, canny_image, num_steps, controlnet_conditioning_scale, seed)
+    images = generate_(prompt, negative_prompt, grayscale_image, num_steps, controlnet_conditioning_scale, seed)
 
-    return [canny_image,images[0]]
+    return [grayscale_image,images[0]]
     
 block = gr.Blocks().queue()