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)
|