Update app.py

app.py

@@ -54,15 +54,14 @@ def generate_(prompt, negative_prompt, grayscale_image, num_steps, controlnet_co
 
 @spaces.GPU
 def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditioning_scale, seed):
-    
     # resize input_image to 1024x1024
     input_image = resize_image(input_image)
-    
+
     grayscale_image = input_image.convert('L').convert('RGB')
-  
     images = generate_(prompt, negative_prompt, grayscale_image, num_steps, controlnet_conditioning_scale, seed)
 
-    return [grayscale_image,images[0]]
+    # Return both grayscale and output images in a list for the gallery
+    return [grayscale_image, images[0]]
     
 block = gr.Blocks().queue()