Spaces:
Sleeping
Sleeping
Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
|
@@ -165,7 +165,7 @@ class InferenceRunner(StateManager):
|
|
| 165 |
self.update_prev_state()
|
| 166 |
|
| 167 |
|
| 168 |
-
st.session_state.button_label = "Reload Model" if self.is_model_loaded and st.session_state.
|
| 169 |
|
| 170 |
with self.col1:
|
| 171 |
if st.session_state.method == "Fine-Tuned Model":
|
|
|
|
| 165 |
self.update_prev_state()
|
| 166 |
|
| 167 |
|
| 168 |
+
st.session_state.button_label = "Reload Model" if self.is_model_loaded and st.session_state.kbvqa.detection_model != st.session_state['detection_model'] else "Load Model"
|
| 169 |
|
| 170 |
with self.col1:
|
| 171 |
if st.session_state.method == "Fine-Tuned Model":
|