rerank model

RAG/colpali.py

@@ -65,6 +65,15 @@ runtime = boto3.client("sagemaker-runtime",aws_access_key_id=st.secrets['user_ac
 
 # Prepare your payload (e.g., text-only input)
 
+if 'top_img' not in st.session_state:
+    st.session_state['top_img'] = ""
+if 'query_token_vectors' not in st.session_state:
+    st.session_state['query_token_vectors'] = ""
+if 'query_tokens' not in st.session_state:
+    st.session_state['query_tokens'] = ""
+
+
+        
 
 @st.cache_resource
 def call_nova(