Spaces:
Runtime error
Runtime error
Update final_book_retriever.py
Browse files- final_book_retriever.py +4 -4
final_book_retriever.py
CHANGED
@@ -38,14 +38,14 @@ def answer(query, min_similarity=0.7):
|
|
38 |
# Retrieve document details for documents with similarity score greater than min_similarity
|
39 |
ranked_documents = []
|
40 |
for index in ranked_indices:
|
41 |
-
similarity_score = similarities[0][index]
|
42 |
if similarity_score > min_similarity:
|
43 |
document_details = {
|
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 |
-
|
49 |
}
|
50 |
ranked_documents.append(document_details)
|
51 |
else:
|
@@ -54,7 +54,7 @@ def answer(query, min_similarity=0.7):
|
|
54 |
|
55 |
if not ranked_documents:
|
56 |
print("No similar books found")
|
57 |
-
|
58 |
return ranked_documents
|
59 |
|
60 |
# Example usage
|
|
|
38 |
# Retrieve document details for documents with similarity score greater than min_similarity
|
39 |
ranked_documents = []
|
40 |
for index in ranked_indices:
|
41 |
+
similarity_score = round(similarities[0][index], 2) # Round similarity score to two decimal places
|
42 |
if similarity_score > min_similarity:
|
43 |
document_details = {
|
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:
|
|
|
54 |
|
55 |
if not ranked_documents:
|
56 |
print("No similar books found")
|
57 |
+
|
58 |
return ranked_documents
|
59 |
|
60 |
# Example usage
|