change file directories

RAG/colpali.py

@@ -159,7 +159,7 @@ def img_highlight(img,batch_queries,query_tokens):
     img_name = os.path.basename(img)  # e.g., "my_image.png"
 
     # Construct the search pattern
-    search_pattern = f"/home/user/app/similarity_maps/similarity_map_{img_name}_token_*"
+    search_pattern = f"/data/app_bkp/similarity_maps/similarity_map_{img_name}_token_*"
 
     # Search for matching files
     matching_files = glob.glob(search_pattern)
@@ -251,7 +251,7 @@ def img_highlight(img,batch_queries,query_tokens):
     for idx, (fig, ax) in enumerate(plots):
         if(idx<3):
             continue
-        savepath = "/home/user/app/similarity_maps/similarity_map_"+(img.split("/"))[-1]+"_token_"+str(idx)+"_"+query_tokens_from_model[idx]+".png"
+        savepath = "/data/app_bkp/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}`")
@@ -344,7 +344,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/user/app/vs/"+final_docs_sorted_20[0][0]['image']
+        img = "/data/app_bkp/vs/"+final_docs_sorted_20[0][0]['image']
         ans = generate_ans(img,query)
             
         images_highlighted = [{'file':img}]