Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -250,7 +250,7 @@ class InferenceRunner(StateManager):
         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
-                               st.session_state.method != st.session_state.previous_state.method
+                               st.session_state[['method'] != st.session_state['previous_state']['method']
             else "Load Model"
         )
 
@@ -292,6 +292,10 @@ class InferenceRunner(StateManager):
                     st.session_state['time_taken_to_load_model'] = int(time.time() - t1)
                     st.session_state['loading_in_progress'] = False
                     st.session_state['model_loaded'] = True
+                    
+            st.write(st.session_state['method'])
+            st.write(st.session_state['previous_state']['method'])
+            
             elif st.session_state.method == "Vision-Language Embeddings Alignment":
                 self.col1.warning(
                     f'Model using {st.session_state.method} is desgined but requires large scale data and multiple '