Update my_model/state_manager.py

my_model/state_manager.py

@@ -137,7 +137,7 @@ class StateManager:
             st.session_state['kbvqa'] = prepare_kbvqa_model()
             st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
             # 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}
+            st.session_state['previous_state'] = {'method': st.session_state.method, 'detection_model': st.session_state.detection_model}
             st.session_state['model_loaded'] = True
             st.session_state['button_label'] = "Reload Model"
             free_gpu_resources()
@@ -225,22 +225,11 @@ class StateManager:
           
                 st.write(st.session_state['detection_model'] == st.session_state['previous_state']['detection_model'])
                 st.write(st.session_state['method'] == st.session_state['previous_state']['method'])
-                st.write(st.session_state['confidence_level'] != st.session_state['previous_state']['confidence_level'])
-
-                # check if only confidence is changed
-                if st.session_state['detection_model'] == st.session_state['previous_state']['detection_model']\
-                    and st.session_state['method'] == st.session_state['previous_state']['method'] \
-                    and st.session_state['confidence_level'] != st.session_state['previous_state']['confidence_level']:
-                     
-                        st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
-                        st.write("reload func", st.session_state['kbvqa'].detection_confidence )
-                        st.session_state['confidence_level_changed'] = True
-                        return  # only update the confidence level
                     
                 prepare_kbvqa_model(only_reload_detection_model=True)
                 st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
                 self.col1.success("Model reloaded with updated settings and ready for inference.")
-                st.session_state['previous_state'] = {'method': st.session_state.method, 'detection_model': st.session_state.detection_model, 'confidence_level': st.session_state.confidence_level}
+                st.session_state['previous_state'] = {'method': st.session_state.method, 'detection_model': st.session_state.detection_model}
                 st.session_state['button_label'] = "Reload Model"
 
             free_gpu_resources()
@@ -292,7 +281,7 @@ class StateManager:
             tuple: A tuple containing the generated caption, detected objects string, and image with bounding boxes.
         """
         img = copy.deepcopy(image)
-        st.text("Analyzing the image .. ")
+        st.text("Analyzing the image .. please wait")
         caption = st.session_state['kbvqa'].get_caption(img)
         image_with_boxes, detected_objects_str = st.session_state['kbvqa'].detect_objects(img)