Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -42,10 +42,11 @@ class StateManager:
 
     def display_model_settings(self):
         st.write("### Current Model Settings:")
-        st.table(pd.DataFrame(st.session_state['model_settings'], index=[0]))
+        data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa']]
+        st.table(pd.DataFrame(data))
 
     def display_session_state(self):
-        st.write("### Current Session State:")
+        st.write("### Current Model:")
         data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items()]
         df = pd.DataFrame(data)
         st.table(df)