Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -301,7 +301,8 @@ class InferenceRunner(StateManager):
 
             st.write(st.session_state['method'])
             st.write(st.session_state['previous_state']['method'])
-            st.write(st.session_state['kbvqa'].kbvqa_model_name)
+            if st.session_state['kbvqa'] is not None:
+                st.write(st.session_state['kbvqa'].kbvqa_model_name)
             
         if self.is_model_loaded:
             free_gpu_resources()