Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -6,8 +6,6 @@ from my_model.KBVQA import KBVQA, prepare_kbvqa_model
 
 
 class StateManager:
- #   def __init__(self):
-  #      self.initialize_state()
 
     def initialize_state(self):
         if 'images_data' not in st.session_state:
@@ -38,7 +36,7 @@ class StateManager:
         st.table(pd.DataFrame(data))
 
     def display_session_state(self):
-        st.write("### Current Model:")
+        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)
@@ -52,7 +50,6 @@ class StateManager:
             st.text("AAAAAAA")
             st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
             #self.update_model_settings(detection_model, confidence_level)
-            st.text("Model is ready for inference.")
             # Update the previous state with current session state values
             st.session_state['previous_state'] = {'method': st.session_state.method, 'detection_model': st.session_state.detection_model, 'confidence_level': st.session_state.confidence_level}