fixing ax fig error

app.py

@@ -27,7 +27,8 @@ def read_result_from_hub(filename):
     return local_path
 
 def make_boundary_plot(boundary):
-    boundary_plot = visualization.plot_boundary(boundary)
+    boundary_ax = visualization.plot_boundary(boundary)
+    boundary_plot = boundary_ax.get_figure()
     return boundary_plot
 
 def make_interactive_plot(boundary):
@@ -89,10 +90,11 @@ def gradio_interface() -> gr.Blocks:
                 n_field_periods = gr.Number(label="Number of Period Fields", value=3)
                 generate_btn = gr.Button(value="Generate")
 
-            boundary_plot = gr.Plot()
-            interactive_plot = gr.Plot()
-            boozer_plot = gr.Plot()
-            flux_surface_plot = gr.Plot()
+            with gr.Column(visible=True):
+                boundary_plot = gr.Plot()
+                interactive_plot = gr.Plot()
+                boozer_plot = gr.Plot()
+                flux_surface_plot = gr.Plot()
 
         def update_ui(mode):
             return (