Update app.py

app.py

@@ -326,13 +326,13 @@ with gr.Blocks(css=custom_css, title="Jewellery Photography Preview") as demo:
             info = gr.Markdown(value="...")
             with gr.Row():
                 run_btn = gr.Button("🎯 Generate", elem_id="button", variant="primary")
-            with gr.Row():
-                with gr.Column():
-                    info1 = gr.Markdown(value="### Detection overlay")
-                    out_overlay = gr.Image(height=400)
-                with gr.Column():
+            with gr.tabs():
+                with gr.TabItem():
                     info2 = gr.Markdown(value="### Final result")
                     out_bg = gr.Image(height=400)
+                with gr.TabItem():
+                    info1 = gr.Markdown(value="### Detection overlay")
+                    out_overlay = gr.Image(height=400)
             out_status = gr.Text(label="Status", interactive=False)
         # ──────── Footer ────────
     gr.HTML("""