Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -113,7 +113,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"
+    button_label = "Reload Model" if st.session_state['kbvqa'] is not None else "Load Model"
     state_manager.display_session_state()
     if st.session_state.method == "Fine-Tuned Model":
         if st.button(button_label):