Update app.py

app.py

@@ -224,18 +224,22 @@ with demo:
             btn = gr.Button("Run model", variant="primary")
 
     with gr.Box():
+        gr.Markdown("<h2><center>Outputs</center></h2>")
         with gr.Row():
             with gr.Column():
-                gr.Markdown("<h2><center>Text output</center></h2>")
-                gr.Markdown("Will only show contiguous text from text blocks, titles and lists")
-                image_text = gr.Textbox()
-                gr.Markdown("<h2><center>First table</center></h2>")
-                html = gr.HTML()
-                gr.Markdown("<h2><center>JSON output</center></h2>") 
-                json = gr.JSON()
+                with gr.Box():
+                    gr.Markdown("<center><strong>Contiguous text</strong></center>")
+                    image_text = gr.Textbox()
+                with gr.Box():
+                    gr.Markdown("<center><strong>Table</strong></center>")
+                    html = gr.HTML()
+                with gr.Box():
+                    gr.Markdown("<center><strong>JSON</strong></center>") 
+                    json = gr.JSON()
             with gr.Column():
-                gr.Markdown("<h2><center>Layout detection</center></h2>")
-                image_output = gr.Image(type="numpy", label="Output Image")
+                with gr.Box():
+                    gr.Markdown("<center><strong>Layout detection</strong></center>")
+                    image_output = gr.Image(type="numpy", label="Output Image")
 
     btn.click(fn=analyze_image, inputs=[inputs, inputs_pdf, tok_input], outputs=[image_output, image_text, html, json])