Spaces:
Runtime error
Runtime error
Update final_book_retriever.py
Browse files- 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)
|