sofarikasid commited on
Commit
1e64423
1 Parent(s): a320e58

Synced repo using 'sync_with_huggingface' Github Action

Browse files
Files changed (1) hide show
  1. mylib/search.py +3 -3
mylib/search.py CHANGED
@@ -26,12 +26,12 @@ llm = CTransformers(
26
 
27
  # Initialize the QA chain once
28
  # qa = RetrievalQA.from_chain_type(llm=llm, chain_type='stuff', retriever=db.as_retriever(search_kwargs={'k': 10}), return_source_documents=True)
29
- #retriever = db.as_retriever(search_kwargs={"k": 10})
30
 
31
 
32
  def final_result(query):
33
- docs_and_scores = db.similarity_search_with_score(query,k=10)
34
- #response = retriever.get_relevant_documents(query)
35
  return docs_and_scores
36
 
37
 
 
26
 
27
  # Initialize the QA chain once
28
  # qa = RetrievalQA.from_chain_type(llm=llm, chain_type='stuff', retriever=db.as_retriever(search_kwargs={'k': 10}), return_source_documents=True)
29
+ # retriever = db.as_retriever(search_kwargs={"k": 10})
30
 
31
 
32
  def final_result(query):
33
+ docs_and_scores = db.similarity_search_with_score(query, k=10)
34
+ # response = retriever.get_relevant_documents(query)
35
  return docs_and_scores
36
 
37