Spaces:
Sleeping
Sleeping
Update RAG/Retriever.py
Browse files- RAG/Retriever.py +2 -2
RAG/Retriever.py
CHANGED
|
@@ -17,10 +17,10 @@ userdata = {
|
|
| 17 |
}
|
| 18 |
gemini_key = userdata.get("GEMINI_API_KEY")
|
| 19 |
parent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
| 20 |
-
|
| 21 |
# sync
|
| 22 |
# Load docs later
|
| 23 |
-
with open(
|
| 24 |
docs = pickle.load(f)
|
| 25 |
client = chromadb.PersistentClient(path=parent_dir)
|
| 26 |
|
|
|
|
| 17 |
}
|
| 18 |
gemini_key = userdata.get("GEMINI_API_KEY")
|
| 19 |
parent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
| 20 |
+
pkl_path = os.path.join(parent_dir, 'split_docs.pkl')
|
| 21 |
# sync
|
| 22 |
# Load docs later
|
| 23 |
+
with open(pkl_path, 'rb') as f:
|
| 24 |
docs = pickle.load(f)
|
| 25 |
client = chromadb.PersistentClient(path=parent_dir)
|
| 26 |
|