Amna2024 commited on
Commit
697b441
·
verified ·
1 Parent(s): a6353f0

Update RAG/Retriever.py

Browse files
Files changed (1) hide show
  1. 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.dirname(os.path.abspath(__file__)))
20
  pkl_path = os.path.join(parent_dir, 'split_docs.pkl')
21
- print('pkl_path:',pkl_path)
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: