Update app.py

app.py

@@ -130,10 +130,12 @@ def run_inference():
     settings_changed = (st.session_state['model_settings']['detection_model'] != detection_model or
                         st.session_state['model_settings']['confidence_level'] != confidence_level)
 
-    if settings_changed:
+    need_model_reload = settings_changed and 'kbvqa' in st.session_state and st.session_state['kbvqa'] is not None
+    
+    if need_model_reload:
         st.text("Model Settings have changed, please reload the model, this will take no time :)")
 
-    button_label = "Reload Model" if settings_changed else "Load Model"
+    button_label = "Reload Model" if need_model_reload else "Load Model"
 
     if method == "Fine-Tuned Model":
         if 'kbvqa' not in st.session_state:
@@ -166,7 +168,7 @@ def run_inference():
             image_qa_app(st.session_state['kbvqa'])
 
     else:
-        st.write('Model is not ready for inference yet')
+        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: