Spaces:
Sleeping
Sleeping
Update rag_utils.py
Browse files- 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)
|