Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ def get_top_chunks(query, chunk_embeddings, text_chunks):
|
|
| 39 |
# Normalize the query embedding to unit length for accurate similarity comparison
|
| 40 |
query_embedding_normalized = query_embedding / query_embedding.norm()
|
| 41 |
|
| 42 |
-
|
| 43 |
chunk_embeddings_normalized = chunk_embeddings / chunk_embeddings.norm(dim=1, keepdim=True)
|
| 44 |
|
| 45 |
# Calculate cosine similarity between query and all chunks using matrix multiplication
|
|
|
|
| 39 |
# Normalize the query embedding to unit length for accurate similarity comparison
|
| 40 |
query_embedding_normalized = query_embedding / query_embedding.norm()
|
| 41 |
|
| 42 |
+
# Normalize all chunk embeddings to unit length for consistent comparison
|
| 43 |
chunk_embeddings_normalized = chunk_embeddings / chunk_embeddings.norm(dim=1, keepdim=True)
|
| 44 |
|
| 45 |
# Calculate cosine similarity between query and all chunks using matrix multiplication
|