Update my_model/state_manager.py

my_model/state_manager.py

@@ -427,12 +427,12 @@ class StateManager:
         return self.col3.write(df)
 
     
-    def display_session_state(self):
+    def display_session_state(self, col):
         """
         Displays a table of the complete application state..
         """
   
-        st.write("Current Model:")
+        col.write("Current Model:")
         data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items()]
         df = pd.DataFrame(data).reset_index(drop=True)
-        st.write(df)
+        col.write(df)