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