Update my_model/state_manager.py

my_model/state_manager.py

@@ -222,9 +222,14 @@ class StateManager:
             free_gpu_resources()
             if self.is_model_loaded:
                 st.text("tttttt")
+                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['confidence_level']['method']:
+                    and st.session_state['confidence_level'] != st.session_state['previous_state']['confidence_level']:
                         st.text("aaaaa")
                         st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
                         st.text("HHHHHH")