prithvirajpawar commited on
Commit
76ed729
·
1 Parent(s): f24b674

addition of intro_msg

Browse files
Files changed (1) hide show
  1. helpmate_ai.py +2 -2
helpmate_ai.py CHANGED
@@ -119,7 +119,7 @@ def retreive_results(query):
119
  # Query the collection against the user query and return the top 10 results
120
  results = insurance_collection.query(
121
  query_texts=query,
122
- n_results=50
123
  )
124
 
125
  # Store the query in cache_collection as document w.r.t to ChromaDB so that it can be embedded and searched against later
@@ -156,7 +156,7 @@ from sentence_transformers import CrossEncoder, util
156
  cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
157
 
158
  #function to re-rank results using cross-encoder
159
- def rerank_with_cross_encoder(query, results_df, top_k=10):
160
 
161
  # Input (query, response) pairs for each of the top 10 responses received from the semantic search to the cross encoder
162
  # Generate the cross_encoder scores for these pairs
 
119
  # Query the collection against the user query and return the top 10 results
120
  results = insurance_collection.query(
121
  query_texts=query,
122
+ n_results=10
123
  )
124
 
125
  # Store the query in cache_collection as document w.r.t to ChromaDB so that it can be embedded and searched against later
 
156
  cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
157
 
158
  #function to re-rank results using cross-encoder
159
+ def rerank_with_cross_encoder(query, results_df, top_k=3):
160
 
161
  # Input (query, response) pairs for each of the top 10 responses received from the semantic search to the cross encoder
162
  # Generate the cross_encoder scores for these pairs