Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -244,9 +244,6 @@ class InferenceRunner(StateManager):
         reload_detection_model = False
         force_reload_full_model = False
 
-        #if self.is_model_loaded and self.settings_changed:
-        if self.settings_changed:
-            self.col1.warning("Model settings have changed, please reload the model.. ")
         # self.update_prev_state()
         st.session_state.button_label = (
             "Reload Model" if (self.is_model_loaded and
@@ -255,6 +252,9 @@ 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.. ")
 
         with self.col1:
             if st.session_state.method == "7b-Fine-Tuned Model" or st.session_state.method == "13b-Fine-Tuned Model":