Update app.py

app.py

@@ -165,19 +165,16 @@ def run_inference():
                 
 
         if st.session_state['kbvqa']:
+            display_model_settings()
             image_qa_app(st.session_state['kbvqa'])
 
     else:
         st.write('Model is not ready yet, will be updated later.')
 
-    # Display model settings
-    if 'kbvqa' in st.session_state and st.session_state['kbvqa'] is not None:
-        model_settings = {
-            'Detection Model': st.session_state['model_settings']['detection_model'],
-            'Confidence Level': st.session_state['model_settings']['confidence_level']
-        }
-        st.write("### Current Model Settings:")
-        st.table(pd.DataFrame(model_settings, index=[0]))
+
+def display_model_settings():
+    st.write("### Current Model Settings:")
+    st.table(pd.DataFrame(st.session_state['model_settings'], index=[0]))