Programmes commited on
Commit
7c15719
·
verified ·
1 Parent(s): da32198

Update rag_utils.py

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