m7mdal7aj commited on
Commit
3731a94
·
verified ·
1 Parent(s): 33ddd8a

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +2 -1
my_model/tabs/run_inference.py CHANGED
@@ -245,7 +245,7 @@ class InferenceRunner(StateManager):
245
  force_reload_full_model = False
246
 
247
  if self.is_model_loaded and self.settings_changed:
248
- self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
249
  # self.update_prev_state()
250
  st.session_state.button_label = (
251
  "Reload Model" if (self.is_model_loaded and
@@ -269,6 +269,7 @@ class InferenceRunner(StateManager):
269
  load_fine_tuned_model = True
270
  elif st.session_state.button_label == "Reload Model" and st.session_state['method'] != \
271
  st.session_state['previous_state']['method']: # check if the model size have changed
 
272
  force_reload_full_model = True
273
  elif (self.is_model_loaded and st.session_state.kbvqa.detection_model !=
274
  st.session_state['detection_model']):
 
245
  force_reload_full_model = False
246
 
247
  if self.is_model_loaded and self.settings_changed:
248
+ self.col1.warning("Model settings have changed, please reload the model.. ")
249
  # self.update_prev_state()
250
  st.session_state.button_label = (
251
  "Reload Model" if (self.is_model_loaded and
 
269
  load_fine_tuned_model = True
270
  elif st.session_state.button_label == "Reload Model" and st.session_state['method'] != \
271
  st.session_state['previous_state']['method']: # check if the model size have changed
272
+ self.col1.warning("Model settings have changed, please reload the model.. ")
273
  force_reload_full_model = True
274
  elif (self.is_model_loaded and st.session_state.kbvqa.detection_model !=
275
  st.session_state['detection_model']):