Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -250,7 +250,8 @@ 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['previous_state']['method'] is not None and 
+                                st.session_state['method'] != st.session_state['previous_state']['method'])
             else "Load Model"
         )