Bofandra commited on
Commit
476fb9f
1 Parent(s): f037bd4

Update app.py

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