fix image selection

app.py

@@ -299,11 +299,10 @@ with gr.Blocks() as demo:
         outputs=[generated_images]
     )
 
-    generated_images.select(
-        fn=lambda x: x,
-        inputs=[generated_images],
-        outputs=[selected_image]
-    )
+    def select_image(evt: gr.SelectData):
+        return evt.value['image']['url']
+
+    generated_images.select(select_image, generated_images, [selected_image])
 
     generate_3d_btn.click(fn=check_input_image, inputs=[selected_image]).success(
         fn=preprocess,