Update app.py

app.py

@@ -176,6 +176,9 @@ Generate 768x768 multi-view images from a single image using SDXL <br>
                         value="watermark, ugly, deformed, noisy, blurry, low contrast",
                     )
 
+        def use_orientation(selected_image:gr.SelectData):
+            return selected_image.value
+
         with gr.Column():
             result = gr.Gallery(
                 label="Result",
@@ -186,6 +189,11 @@ Generate 768x768 multi-view images from a single image using SDXL <br>
                 height="auto",
             )
 
+            if result:
+                input_fg = gr.Image(type="numpy", label="Selected Image", height=480)
+                result.select(use_orientation, result, input_fg)
+
+
     # with gr.Row():
         # gr.Examples(
         #     examples=examples,