onyiokoye commited on
Commit
c73aebd
·
verified ·
1 Parent(s): 5354d75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=3).indices
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)