qsaheeb
commited on
Commit
·
54a9816
1
Parent(s):
9d426bd
Final changes
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ with open("model/sbert_embeddings2.pkl", "rb") as f:
|
|
15 |
|
16 |
# Load models
|
17 |
retriever_model = SentenceTransformer("all-mpnet-base-v2") # More accurate than MiniLM
|
18 |
-
reranker_model = CrossEncoder("cross-encoder/
|
19 |
|
20 |
def fetch_summary_duckduckgo(book_title, log):
|
21 |
"""Fetch book summary from DuckDuckGo API (search engine)."""
|
|
|
15 |
|
16 |
# Load models
|
17 |
retriever_model = SentenceTransformer("all-mpnet-base-v2") # More accurate than MiniLM
|
18 |
+
reranker_model = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2") # More accurate ranking
|
19 |
|
20 |
def fetch_summary_duckduckgo(book_title, log):
|
21 |
"""Fetch book summary from DuckDuckGo API (search engine)."""
|