tryout

streamlit_simulation/app.py

@@ -183,11 +183,10 @@ def predict_transformer_step(model, dataset, idx, device):
 
 
 def init_simulation_layout():
-    col1, spacer, col2 = st.columns([3, 0.2, 1])
-    plot_title = col1.empty()
-    plot_container = col1.empty()
-    x_axis_label = col1.empty()
-    info_container = col2.empty()
+    plot_title = st.empty()
+    plot_container = st.empty()
+    x_axis_label = st.empty()
+    info_container = st.empty()
     return plot_title, plot_container, x_axis_label, info_container