Update app.py

app.py

@@ -144,7 +144,7 @@ with gr.Blocks() as demo:
         message_box = gr.Textbox(interactive=False, label="Message")
 
     # Dataset Viewer and Pagination Controls at the Bottom
-    with gr.Column(visible=False):
+    with gr.Column(visible=False) as dataset_viewer_section:
         gr.Markdown("### Dataset Viewer")
         dataset_html = gr.HTML()
         with gr.Row():