huynhdoo commited on
Commit
c0bfa48
1 Parent(s): 6cfc71b

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -148,6 +148,7 @@ def retrieve(queries=['je veux bricoler'], origin='Formation', n_results=10):
148
 
149
  filter = ['score', 'key', 'label']
150
  res = res[filter].groupby(by=['key', 'label']).max().sort_values(by=['score'], ascending=False).reset_index()
 
151
  return res[filter]
152
 
153
  def rank(query='cuisine', documents = []):
 
148
 
149
  filter = ['score', 'key', 'label']
150
  res = res[filter].groupby(by=['key', 'label']).max().sort_values(by=['score'], ascending=False).reset_index()
151
+ res['score'] = res['score'].apply(lambda x: round(x, 3))
152
  return res[filter]
153
 
154
  def rank(query='cuisine', documents = []):