Revert back gr.Image in gallery

app.py

@@ -8,6 +8,7 @@ import numpy as np
 import spaces
 import torch
 from diffusers import FluxFillPipeline
+from gradio.components.gallery import GalleryMediaType
 from gradio.components.image_editor import EditorValue
 from PIL import Image, ImageFilter, ImageOps
 
@@ -28,7 +29,7 @@ if not torch.cuda.is_available():
     def _dummy_pipe(image: Image.Image, *args, **kwargs):  # noqa: ARG001
         # return {"images": [image]}  # noqa: ERA001
         blue_image = Image.new("RGB", image.size, (0, 0, 255))
-        return {"images": [blue_image]}
+        return {"images": [blue_image, blue_image, blue_image]}
 
     pipe = _dummy_pipe
 else:
@@ -185,7 +186,7 @@ def infer(
     max_dimension: int = 720,
     num_images_per_prompt: int = 2,
     progress: gr.Progress = gr.Progress(track_tqdm=True),  # noqa: ARG001, B008
-):
+) -> tuple[GalleryMediaType, int]:
     # Ensure max_dimension is a multiple of 16 (for VAE)
     max_dimension = (max_dimension // 16) * 16
 
@@ -317,15 +318,6 @@ def infer(
             room_mask_cropped,
         )
         final_images.append(final_image)
-    final_images_gradio = [
-        gr.Image(
-            value=image,
-            label=f"Result {i + 1}",
-            show_label=False,
-            interactive=False,
-        )
-        for i, image in enumerate(final_images)
-    ]
 
     return final_images, seed
 
@@ -447,6 +439,7 @@ with gr.Blocks(css=css) as demo:
                 show_label=False,
                 columns=2,
                 height=500,
+                interactive=False,
             )
             run_button = gr.Button("Run")