Bofandra commited on
Commit
592fc4e
1 Parent(s): 320811f

Update app.py

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