Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
|
@@ -114,7 +114,7 @@ def run_inference():
|
|
| 114 |
# need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
|
| 115 |
|
| 116 |
button_label = "Reload Model" if st.session_state['kbvqa'] else "Load Model"
|
| 117 |
-
|
| 118 |
if st.session_state.method == "Fine-Tuned Model":
|
| 119 |
if st.button(button_label):
|
| 120 |
if button_label == "Load Model" and state_manager.is_model_loaded():
|
|
|
|
| 114 |
# need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
|
| 115 |
|
| 116 |
button_label = "Reload Model" if st.session_state['kbvqa'] else "Load Model"
|
| 117 |
+
state_manager.display_session_state()
|
| 118 |
if st.session_state.method == "Fine-Tuned Model":
|
| 119 |
if st.button(button_label):
|
| 120 |
if button_label == "Load Model" and state_manager.is_model_loaded():
|