Spaces:
Sleeping
Sleeping
Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
@@ -250,7 +250,7 @@ class InferenceRunner(StateManager):
|
|
250 |
st.session_state.button_label = (
|
251 |
"Reload Model" if (self.is_model_loaded and
|
252 |
st.session_state.kbvqa.detection_model != st.session_state['detection_model']) or
|
253 |
-
|
254 |
else "Load Model"
|
255 |
)
|
256 |
|
|
|
250 |
st.session_state.button_label = (
|
251 |
"Reload Model" if (self.is_model_loaded and
|
252 |
st.session_state.kbvqa.detection_model != st.session_state['detection_model']) or
|
253 |
+
st.session_state.method != st.session_state.previous_state.method
|
254 |
else "Load Model"
|
255 |
)
|
256 |
|