achdaisy commited on
Commit
530f8f8
1 Parent(s): 732464a

Update final_book_retriever.py

Browse files
Files changed (1) hide show
  1. final_book_retriever.py +5 -5
final_book_retriever.py CHANGED
@@ -44,8 +44,8 @@ def answer(query, min_similarity=0.7):
44
  "Book": data['Book Title'].iloc[index],
45
  "Author": data['Author'].iloc[index],
46
  "Edition": data['Edition'].iloc[index],
47
- "File Name": data['File_name'].iloc[index],
48
- "Similarity Score": similarity_score
49
  }
50
  ranked_documents.append(document_details)
51
  else:
@@ -58,6 +58,6 @@ def answer(query, min_similarity=0.7):
58
  return ranked_documents
59
 
60
  # Example usage
61
- query = "machine learning"
62
- result = answer(query)
63
- print(result)
 
44
  "Book": data['Book Title'].iloc[index],
45
  "Author": data['Author'].iloc[index],
46
  "Edition": data['Edition'].iloc[index],
47
+ "File Name": data['File_name'].iloc[index]
48
+ #"Similarity Score": similarity_score
49
  }
50
  ranked_documents.append(document_details)
51
  else:
 
58
  return ranked_documents
59
 
60
  # Example usage
61
+ #query = "machine learning"
62
+ #result = answer(query)
63
+ #print(result)