Update app.py

app.py

@@ -240,7 +240,9 @@ with gr.Blocks() as demo:
     with gr.Row():
         agent_btn = gr.Button("Run AI Agent (5 Insights + 5 Visualizations)")
         insights_output = gr.Textbox(label="Insights from SmolAgent", lines=15)
-        visual_output = gr.Gallery(label="Generated Visualizations").style(grid=3, height="auto")
+        #visual_output = gr.Gallery(label="Generated Visualizations").style(grid=3, height="auto")
+        visual_output = gr.Gallery(label="Generated Visualizations", columns=[3], height=400)
+
 
     #agent_btn.click(fn=run_agent, inputs=df_output, outputs=insights_output)
     agent_btn.click(fn=run_agent, inputs=df_output, outputs=[insights_output, visual_output])