# Calculate similarity scores
scores = np.dot(query_emb, np.transpose(doc_emb))[0]