Update app.py
Browse files
app.py
CHANGED
|
@@ -106,7 +106,7 @@ def get_top_chunks(query, chunk_embeddings, text_chunks):
|
|
| 106 |
print(similarities)
|
| 107 |
|
| 108 |
# Find the indices of the 3 chunks with highest similarity scores
|
| 109 |
-
top_indices = torch.topk(similarities, k=
|
| 110 |
|
| 111 |
# Print the top indices
|
| 112 |
print(top_indices)
|
|
|
|
| 106 |
print(similarities)
|
| 107 |
|
| 108 |
# Find the indices of the 3 chunks with highest similarity scores
|
| 109 |
+
top_indices = torch.topk(similarities, k=5).indices
|
| 110 |
|
| 111 |
# Print the top indices
|
| 112 |
print(top_indices)
|