Upload folder using huggingface_hub
Browse files
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 = []):
|