Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -244,7 +244,6 @@ class InferenceRunner(StateManager):
         reload_detection_model = False
         force_reload_full_model = False
 
-        # self.update_prev_state()
         st.session_state.button_label = (
             "Reload Model" if (self.is_model_loaded and
                                st.session_state.kbvqa.detection_model != st.session_state['detection_model']) or
@@ -252,7 +251,7 @@ class InferenceRunner(StateManager):
                                st.session_state['method'] != st.session_state['previous_state']['method'])
             else "Load Model"
         )
-                #if self.is_model_loaded and self.settings_changed:
+                
         if st.session_state.button_label == "Reload Model":
             self.col1.warning("Model settings have changed, please reload the model.. ")
 
@@ -304,11 +303,6 @@ class InferenceRunner(StateManager):
                     f'Model using {st.session_state.method} is desgined but requires large scale data and multiple '
                     f'high-end GPUs, implementation will be explored in the future.')
 
-            st.write(st.session_state['method'])
-            st.write(st.session_state['previous_state']['method'])
-            if st.session_state['kbvqa'] is not None:
-                st.write(st.session_state['kbvqa'].kbvqa_model_name)
-
         if self.is_model_loaded:
             free_gpu_resources()
             st.session_state['loading_in_progress'] = False