Spaces:
Sleeping
Sleeping
Update actions/search_content.py
Browse files
actions/search_content.py
CHANGED
@@ -54,7 +54,7 @@ def main_search(query):
|
|
54 |
faiss_index = load_faiss_index(FAISS_INDEX_FILE_PATH)
|
55 |
data_frame_indexed = load_data(DATA_FILE_PATH)
|
56 |
results = search_content(query, data_frame_indexed, transformer_model, faiss_index)
|
57 |
-
return results['
|
58 |
|
59 |
if __name__ == "__main__":
|
60 |
query = "school courses"
|
|
|
54 |
faiss_index = load_faiss_index(FAISS_INDEX_FILE_PATH)
|
55 |
data_frame_indexed = load_data(DATA_FILE_PATH)
|
56 |
results = search_content(query, data_frame_indexed, transformer_model, faiss_index)
|
57 |
+
return results['Answers'] # return the 'Answers' column
|
58 |
|
59 |
if __name__ == "__main__":
|
60 |
query = "school courses"
|