Programmes commited on
Commit
3b93dfc
·
verified ·
1 Parent(s): e5d350e

Update rag_utils.py

Browse files
Files changed (1) hide show
  1. rag_utils.py +1 -1
rag_utils.py CHANGED
@@ -6,7 +6,7 @@ import re
6
  from sentence_transformers import SentenceTransformer
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
9
- def load_faiss_index(index_path="faiss_index.faiss", doc_path="documents.pkl"):
10
  index = faiss.read_index(index_path)
11
  with open(doc_path, "rb") as f:
12
  documents = pickle.load(f)
 
6
  from sentence_transformers import SentenceTransformer
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
9
+ def load_faiss_index(index_path="faiss_index/faiss_index.faiss", doc_path="faiss_index/documents.pkl"):
10
  index = faiss.read_index(index_path)
11
  with open(doc_path, "rb") as f:
12
  documents = pickle.load(f)