RAG changes

.gitignore

@@ -1,3 +1,4 @@
 **/__pycache__/
 *.DS_Store
-vs
+vs
+similarity_maps

RAG/colpali.py

@@ -234,7 +234,7 @@ def colpali_search_rerank(query):
             final_docs.append(doc)
         final_docs_sorted = sorted(final_docs, key=lambda d: d['total_score'], reverse=True)
         final_docs_sorted_20.append(final_docs_sorted[:20])
-        img = "/home/ubuntu/AI-search-with-amazon-opensearch-service/vs/"+final_docs_sorted_20[0][0]['image']
+        img = "/home/user/app/vs/"+final_docs_sorted_20[0][0]['image']
         ans = generate_ans(img,query)
         images_highlighted = [{'file':img}]
         # if(st.session_state.show_columns == True):
@@ -330,7 +330,7 @@ def img_highlight(img,batch_queries,query_tokens):
     for idx, (fig, ax) in enumerate(plots):
         if(idx<3):
             continue
-        savepath = "/home/ubuntu/AI-search-with-amazon-opensearch-service/similarity_maps/similarity_map_"+(img.split("/"))[-1]+"_token_"+str(idx)+"_"+query_tokens_from_model[idx]+".png"
+        savepath = "/home/user/app/similarity_maps/similarity_map_"+(img.split("/"))[-1]+"_token_"+str(idx)+"_"+query_tokens_from_model[idx]+".png"
         fig.savefig(savepath, bbox_inches="tight")
         map_images.append({'file':savepath})
         print(f"Similarity map for token `{query_tokens_from_model[idx]}` saved at `{savepath}`")