pvanand commited on
Commit
add5800
·
1 Parent(s): ea8523f

Update actions/search_content.py

Browse files
Files changed (1) hide show
  1. actions/search_content.py +1 -1
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['QNA'] # return the 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"