Update app.py

app.py

@@ -31,17 +31,27 @@ sample_images = ["Files/sample1.jpg", "Files/sample2.jpg", "Files/sample3.jpg",
 def run_inference():
     st.title("Run Inference")
 
+    # Initialize session state for the model
+    if 'kbvqa' not in st.session_state:
+        st.session_state['kbvqa'] = None
+
     # Button to load KBVQA models
     if st.button('Load KBVQA Models'):
-        # Call the function to load models and show progress
-        kbvqa = prepare_kbvqa_model('yolov5')  
+        if st.session_state['kbvqa'] is not None:
+            st.write("Model already loaded.")
+        else:
+            # Call the function to load models and show progress
+            st.session_state['kbvqa'] = prepare_kbvqa_model('yolov5')  # Replace with your model
 
-        if kbvqa:
+        if st.session_state['kbvqa']:
             st.write("Model is ready for inference.")
-            image_qa_app(kbvqa)
         else:
             st.write("Please load the model first")
 
+    if st.session_state['kbvqa']:
+        image_qa_app(st.session_state['kbvqa'])
+
+
 
 def image_qa_app(kbvqa):
     # Initialize session state for storing the current image and its Q&A history
@@ -123,4 +133,4 @@ def main():
         run_object_detection()
 
 if __name__ == "__main__":
-    main()
+    main()