Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -114,12 +114,13 @@ def run_inference():
             else:
                 state_manager.reload_detection_model()
                 st.success("Model reloaded with updated settings and ready for inference.")
-        st.write(st.session_state.kbvqa)
-        st.write(st.session_state.kbvqa.all_models_loaded)
-        st.write(st.session_state.kbvqa.all_models_loaded())
-        st.write(st.session_state.kbvqa)
-        if state_manager.is_model_loaded() and st.session_state.kbvqa.all_models_loaded:
-            image_qa_app(state_manager.get_model())
+
+            if state_manager.is_model_loaded() and st.session_state.kbvqa.all_models_loaded:
+                image_qa_app(state_manager.get_model())
+
+     
+            st.write(st.session_state.kbvqa.all_models_loaded)
+
     else:
         st.write('Model is not ready yet, will be updated later.')