Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -31,9 +31,11 @@ class StateManager:
                 st.session_state['model_settings']['selected_method'] != current_selected_method)
 
     def display_model_settings(self):
-        st.write("### Current Model Settings:")
+        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']]
-        st.table(pd.DataFrame(data))
+        df = pd.DataFrame(data)
+        styled_df = df.style.set_properties(**{'background-color': 'black', 'color': 'lime', 'border-color': 'white'})
+        st.table(styled_df)
 
     def display_session_state(self):
         st.write("Current Model:")
@@ -47,7 +49,6 @@ class StateManager:
             free_gpu_resources()
             st.text("Loading the model, this should take no more than a few minutes, please wait...")
             st.session_state['kbvqa'] = prepare_kbvqa_model()
-            st.text("AAAAAAA")
             st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
             #self.update_model_settings(detection_model, confidence_level)
             # Update the previous state with current session state values
@@ -95,6 +96,7 @@ class StateManager:
 
     def analyze_image(self, image, kbvqa):
         img = copy.deepcopy(image)
+        st.text("Analyzing the image .. ")
         caption = kbvqa.get_caption(img)
         image_with_boxes, detected_objects_str = kbvqa.detect_objects(img)
         return caption, detected_objects_str, image_with_boxes