Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -84,7 +84,7 @@ class InferenceRunner(StateManager):
                     nested_col21, nested_col22 = st.columns([0.65, 0.35])
                     image_for_display = self.resize_image(image_data['image'], 600)
                     nested_col21.image(image_for_display, caption=f'Uploaded Image: {image_key[-11:]}')
-                    if not image_data['analysis_done'] or self.settings_changed or st.session_state['confidence_level_changed']:  # if not done analysis before or even done but settings changed, then we need to analyze again
+                    if not image_data['analysis_done'] or self.settings_changed or self.confidance_change:  # if not done analysis before or even done but settings changed, then we need to analyze again
                         nested_col22.text("Please click 'Analyze Image'..")
                         free_gpu_resources()
                         with nested_col22:
@@ -100,7 +100,7 @@ class InferenceRunner(StateManager):
                   
                     if image_data['analysis_done']:
                         free_gpu_resources()
-                        if st.session_state['confidence_level_changed']:
+                        if self.confidance_change:
                             nested_col22.warning("Confidence level changed, please click analyze again.")
                         st.session_state['loading_in_progress'] = False
                         sample_questions = config.SAMPLE_QUESTIONS.get(image_key, [])
@@ -148,7 +148,7 @@ class InferenceRunner(StateManager):
         force_reload_full_model = False
         
         #st.session_state['settings_changed'] = self.has_state_changed()
-        if st.session_state['settings_changed']:
+        if self.settings_changed:
             self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
            
         st.session_state.button_label = "Reload Model" if self.is_model_loaded and self.settings_changed else "Load Model"