KB-VQA-E

Sleeping

m7mdal7aj commited on Feb 26

Commit

e490a7c

•

1 Parent(s): c03044f

Update my_model/tabs/run_inference.py

Files changed (1) hide show

my_model/tabs/run_inference.py CHANGED Viewed

@@ -114,7 +114,7 @@ def run_inference():
    # need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
     button_label = "Reload Model" if st.session_state['kbvqa'] else "Load Model"
     if st.session_state.method == "Fine-Tuned Model":
         if st.button(button_label):
             if button_label == "Load Model" and state_manager.is_model_loaded():

    # need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
     button_label = "Reload Model" if st.session_state['kbvqa'] else "Load Model"
+    state_manager.display_session_state()
     if st.session_state.method == "Fine-Tuned Model":
         if st.button(button_label):
             if button_label == "Load Model" and state_manager.is_model_loaded():