Jan Mühlnikel
commited on
Commit
·
e8954be
1
Parent(s):
fd577be
fixed search
Browse files
functions/semantic_search.py
CHANGED
|
@@ -24,4 +24,4 @@ def search(query, model, embeddings, filtered_df, top_x=30):
|
|
| 24 |
#print(filtered_df.columns())
|
| 25 |
top_indecies = [i for i in I[0]]
|
| 26 |
|
| 27 |
-
return filtered_df.iloc[top_indecies]
|
|
|
|
| 24 |
#print(filtered_df.columns())
|
| 25 |
top_indecies = [i for i in I[0]]
|
| 26 |
|
| 27 |
+
return filtered_df.iloc[top_indecies]
|