normalize

app.py

@@ -26,6 +26,7 @@ esm_model = EsmEmbedding()
 
 # --- Helper functions ---
 def get_nn(query_vector, k=16):
+    query_vector = query_vector/np.linalg.norm(query_vector)
     results = pc_index.query(
         vector=query_vector.tolist(),
         top_k=k,