Update app.py

app.py

@@ -143,6 +143,23 @@ with gr.Blocks(css='style.css') as demo:
             </div>
         """
     )
+    with gr.Blocks() as demo:
+    with gr.Row():
+        image_input = gr.Image()
+        # image_output = gr.Image() # download [ok]
+        image_output = gr.Gallery(
+                label='Output',
+                show_label=False,
+                elem_id='gallery'
+            ).style(
+                grid=2,height='auto'
+            )
+    image_button = gr.Button("Flip")
+    image_button.click(
+        flip_image, 
+        inputs=image_input, 
+        outputs=image_output
+    )
     with gr.Column(elem_id="col-container"):
         with gr.Row(variant="compact"):
             input_text = gr.Textbox(