updated output to work with a single image

app.py

@@ -83,7 +83,7 @@ def generate_images(prompt, img_filepath,
         num_samples=n_images, width=512, height=512, num_inference_steps=num_inference_steps, seed=seed,
         scale=img_prompt_scale, # with scale=1 I get weird images
     )
-    return [images, Image.fromarray(face_image[..., [2, 1, 0]])]
+    return [images[0], Image.fromarray(face_image[..., [2, 1, 0]])]
 
 
 with gr.Blocks() as demo:
@@ -91,6 +91,8 @@ with gr.Blocks() as demo:
     """
     # IP-Adapter-FaceID-plus
     Generate images conditioned on a image prompt and a text prompt. Learn more here: https://huggingface.co/h94/IP-Adapter-FaceID
+
+    This demo is intended to use on GPU. It will work also on CPU but generating one image could take 900 seconds compared to a few sceonds on GPU.
     """)
     with gr.Row():
         with gr.Column():
@@ -104,7 +106,7 @@ with gr.Blocks() as demo:
 
         with gr.Column():
             demo_outputs = []
-            demo_outputs.append(gr.Gallery(label='generated images'))
+            demo_outputs.append(gr.Image(label='generated image'))
             demo_outputs.append(gr.Image(label='detected face', height=224, width=224))
     btn.click(generate_images, inputs=demo_inputs, outputs=demo_outputs)
     sample_prompts = [