qsaheeb commited on
Commit
54a9816
·
1 Parent(s): 9d426bd

Final changes

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/all-MiniLM-L6-v2")
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)."""