upd

app.py

@@ -44,7 +44,7 @@ def process_image(input_image: Image.Image, to_grayscale: bool, prompt: str) ->
         num_inference_steps=20,
     )
 
-    return image_grid(output.images, 1, 1)
+    return output.images[0]
 
 image_input = gr.components.Image(label="Input Image")
 grayscale_checkbox = gr.components.Checkbox(label="Convert output to grayscale", default=False)