crodri commited on
Commit
395860b
1 Parent(s): 1cdd61e

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -22,7 +22,7 @@ class RAG:
22
 
23
  # load vectore store
24
  embeddings = HuggingFaceEmbeddings(model_name=embeddings_model, model_kwargs={'device': 'cpu'})
25
- self.vectore_store = FAISS.load_local(vectorstore, embeddings, allow_dangerous_deserialization=True)#, allow_dangerous_deserialization=True)
26
 
27
  logging.info("RAG loaded!")
28
 
 
22
 
23
  # load vectore store
24
  embeddings = HuggingFaceEmbeddings(model_name=embeddings_model, model_kwargs={'device': 'cpu'})
25
+ self.vectore_store = FAISS.load_local("index-intfloat_multilingual-e5-small-500-100-CA-ES", embeddings, allow_dangerous_deserialization=True)#, allow_dangerous_deserialization=True)
26
 
27
  logging.info("RAG loaded!")
28