diff --git a/find_similar_img.py b/find_similar_img.py index cc8e8f5..6df859e 100644 --- a/find_similar_img.py +++ b/find_similar_img.py @@ -6,8 +6,8 @@ from sklearn.metrics.pairwise import cosine_similarity from embedder import ImageEmbedder # ——— Load stored embeddings & mapping ——— -vecs = np.load("processed_images/image_vectors.npy") # (N, D) -with open("processed_images/index_to_file.pkl", "rb") as f: +vecs = np.load("imgVecs/image_vectors.npy") # (N, D) +with open("imgVecs/index_to_file.pkl", "rb") as f: idx2file = pickle.load(f) # dict: idx → filepath # ——— Specify query image ———