fix selected_image

app.py

@@ -228,17 +228,12 @@ with gr.Blocks() as demo:
             with gr.Row():
                 num_images = gr.Slider(
                     label="Number of Images", minimum=1, maximum=8, step=1, value=4, interactive=True)
-                height = gr.Number(label="Image Height",
-                                   value=1024, interactive=True)
-                width = gr.Number(label="Image Width",
-                                  value=1024, interactive=True)
                 prompt = gr.Text(
                     label="Prompt", value="a photo of a cat", interactive=True)
-                seed = gr.Number(label="Seed", value=3413, interactive=True)
                 generate_2d_btn = gr.Button(value="Generate 2D Images")
 
             with gr.Row():
-                generated_images = gr.Gallery(height=1024)
+                generated_images = gr.Gallery(height=512)
 
             with gr.Row():
                 selected_image = gr.Image(
@@ -295,14 +290,14 @@ with gr.Blocks() as demo:
 
     generate_2d_btn.click(
         fn=process_image,
-        inputs=[num_images, height, width, prompt, seed],
+        inputs=[num_images, 512, 512, prompt, 123],
         outputs=[generated_images]
     )
 
     def select_image(evt: gr.SelectData):
         return evt.value['image']['url']
 
-    generated_images.select(select_image, generated_images, [selected_image])
+    generated_images.select(select_image, None, selected_image)
 
     generate_3d_btn.click(fn=check_input_image, inputs=[selected_image]).success(
         fn=preprocess,