Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -13,7 +13,7 @@ class StateManager:
         if 'kbvqa' not in st.session_state:
             st.session_state['kbvqa'] = None
         if "button_lablel" not in st.session_state: 
-            st.session_state['kbvqa'] = "Load Model"
+            st.session_state['button_lablel'] = "Load Model"
         if "previous_state" not in st.session_state:
             st.session_state['previous_state'] = {}
             
@@ -39,7 +39,7 @@ class StateManager:
         st.write("#### Current Model Settings:")
         data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa']]
         df = pd.DataFrame(data)
-        styled_df = df.style.set_properties(**{'background-color': 'black', 'color': 'white', 'border-color': 'white'}).set_table_styles([{'selector': 'th','props': [('background-color', 'gray'), ('font-weight', 'bold')]}])
+        styled_df = df.style.set_properties(**{'background-color': 'black', 'color': 'white', 'border-color': 'white'}).set_table_styles([{'selector': 'th','props': [('background-color', 'white'), ('font-weight', 'bold')]}])
         st.table(styled_df)
 
     def display_session_state(self):