remove num of images

app.py

@@ -45,7 +45,7 @@ def generate_image(
     guidance: float = 3.5,
     i_guidance: float = 1.0,
     t_guidance: float = 1.0,
-    num_images: int = 4,
+    num_images: int = 1,
 ):
     w, h, min_size = image.size[0], image.size[1], min(image.size)
     image = image.crop(
@@ -69,7 +69,7 @@ def generate_image(
         guidance_scale_real_i=i_guidance,
         guidance_scale_real_t=t_guidance,
         gemini_prompt=gemini_prompt,
-    ).images
+    ).images[0]
 
     return result_image
 
@@ -148,10 +148,10 @@ with demo:
             gr.Slider(minimum=1.0, maximum=6.0, step=0.5, value=3.5, label="guidance scale", info="Tip: start with 3.5, then gradually increase if the consistency is consistently off"),
             gr.Slider(minimum=1.0, maximum=2.0, step=0.05, value=1.5, label="real guidance scale for image", info="Tip: increase if the image is not consistent"),
             gr.Slider(minimum=1.0, maximum=2.0, step=0.05, value=1.0, label="real guidance scale for prompt", info="Tip: increase if the prompt is not consistent"),
-            gr.Slider(minimum=1, maximum=5, step=1, value=4, label="Number of images", info="Select how many images to generate"),
+            gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Number of images", info="Select how many images to generate"),
         ],
-        # outputs=gr.Image(type="pil"),
-        outputs=gr.Gallery(label="Generated Images", height=544),
+        outputs=gr.Image(type="pil"),
+        # outputs=gr.Gallery(label="Generated Images", height=544),
         # examples=get_samples(),
         live=False,
     )