Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -110,6 +110,7 @@ def run_inference():
     if method == "Fine-Tuned Model":
         if st.button(button_label):
             if not need_model_reload:
+                st write(st.session_state['kbvqa'])
                 st.write("Model already loaded.")
             else:
                 st.text("Loading the model, please wait...")