Spaces:

sabazo
/

insurance_advisor_wb

Sleeping

isayahc commited on Jul 18, 2024

Commit

0e17a46

unverified ·

1 Parent(s): 6f22fc5

refactoring get_db_retriever

Files changed (1) hide show

rag_app/get_db_retriever.py CHANGED Viewed

@@ -16,8 +16,6 @@ from config import EMBEDDING_MODEL
 def get_db_retriever(vector_db:str=None):
     embeddings = HuggingFaceHubEmbeddings(repo_id=EMBEDDING_MODEL)
-    #db = Chroma(persist_directory="./vectorstore/lc-chroma-multi-mpnet-500", embedding_function=embeddings)
-    #db.get()
     if not vector_db:
         FAISS_INDEX_PATH='./vectorstore/py-faiss-multi-mpnet-500'
     else:

 def get_db_retriever(vector_db:str=None):
     embeddings = HuggingFaceHubEmbeddings(repo_id=EMBEDDING_MODEL)
     if not vector_db:
         FAISS_INDEX_PATH='./vectorstore/py-faiss-multi-mpnet-500'
     else: