Hi @ravijoe, as suggested earlier in the thread, you could try using FAISS for the similarity search on the BERT embeddings. There’s many tutorials online, eg https://towardsdatascience.com/how-to-build-a-semantic-search-engine-with-transformers-and-faiss-dcbea307a0e8 28