Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -241,8 +241,10 @@ class InferenceRunner(StateManager):
 
         load_fine_tuned_model = False
         fine_tuned_model_already_loaded = False
+        reload_kbvqa = False
         reload_detection_model = False
         force_reload_full_model = False
+        
 
         if self.is_model_loaded and self.settings_changed:
             self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
@@ -267,7 +269,10 @@ class InferenceRunner(StateManager):
                                 fine_tuned_model_already_loaded = True
                             else:
                                 load_fine_tuned_model = True
-                        else:
+                        elif st.session_state.button_label == "Reload Model" 
+                            and st.session_state['method'] != st.session_state['previous_state']['method']: # check if the model size have changed
+                            force_reload_full_model = True
+                        elif self.is_model_loaded and st.session_state.kbvqa.detection_model != st.session_state['detection_model']):
                             reload_detection_model = True
                     if nested_col12.button("Force Reload", on_click=self.disable_widgets,
                                            disabled=self.is_widget_disabled):