Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
@@ -272,7 +272,7 @@ class InferenceRunner(StateManager):
|
|
272 |
elif st.session_state.button_label == "Reload Model"
|
273 |
and st.session_state['method'] != st.session_state['previous_state']['method']: # check if the model size have changed
|
274 |
force_reload_full_model = True
|
275 |
-
elif self.is_model_loaded and st.session_state.kbvqa.detection_model != st.session_state['detection_model']):
|
276 |
reload_detection_model = True
|
277 |
if nested_col12.button("Force Reload", on_click=self.disable_widgets,
|
278 |
disabled=self.is_widget_disabled):
|
|
|
272 |
elif st.session_state.button_label == "Reload Model"
|
273 |
and st.session_state['method'] != st.session_state['previous_state']['method']: # check if the model size have changed
|
274 |
force_reload_full_model = True
|
275 |
+
elif (self.is_model_loaded and st.session_state.kbvqa.detection_model != st.session_state['detection_model']):
|
276 |
reload_detection_model = True
|
277 |
if nested_col12.button("Force Reload", on_click=self.disable_widgets,
|
278 |
disabled=self.is_widget_disabled):
|