Update app.py

app.py

@@ -103,7 +103,7 @@ def run_inference():
     if model_changed or confidence_changed:
         st.session_state['detection_model'] = detection_model
         st.session_state['confidence_level'] = confidence_level
-        st.warning("Detection model or confidence level changed. Please reload the model.")
+        st.warning("Detection model or confidence level changed. Please reload the model, this will take few seconds :)")
 
 
     
@@ -114,20 +114,7 @@ def run_inference():
             st.session_state['kbvqa'] = None
     
         # Button to load KBVQA models
-        if st.button('Load Model'):
-            if st.session_state['kbvqa'] is not None:
-                st.write("Model already loaded.")
-            else:
-                # Call the function to load models and show progress
-                st.text("Loading the model will take no more than a few minutes . .")
-                st.session_state['kbvqa'] = prepare_kbvqa_model(detection_model)
-                st.session_state['kbvqa'].detection_confidence = confidence_level
-    
-            if st.session_state['kbvqa']:
-                st.write("Model is ready for inference.")
-    
-        if st.session_state['kbvqa']:
-            image_qa_app(st.session_state['kbvqa'])
+
 
 
         if st.button('Load Model'):