Bofandra commited on
Commit
4f303fe
·
verified ·
1 Parent(s): 5d4a7cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def respond(
46
  scores = (query_embeddings @ encoded_questions.T) * 100
47
  selected_references['similarity'] = scores.tolist()[0]
48
  sorted_references = selected_references.sort_values(by='similarity', ascending=False)
49
- sorted_references = sorted_references.iloc[:1]
50
  sorted_references = sorted_references.sort_values(by='similarity', ascending=True)
51
  print(sorted_references.shape[0])
52
  print(sorted_references['similarity'].tolist())
 
46
  scores = (query_embeddings @ encoded_questions.T) * 100
47
  selected_references['similarity'] = scores.tolist()[0]
48
  sorted_references = selected_references.sort_values(by='similarity', ascending=False)
49
+ sorted_references = sorted_references.iloc[:3]
50
  sorted_references = sorted_references.sort_values(by='similarity', ascending=True)
51
  print(sorted_references.shape[0])
52
  print(sorted_references['similarity'].tolist())