Amna2024 commited on
Commit
a03948e
·
verified ·
1 Parent(s): b09d763

Update RAG/Retriever.py

Browse files
Files changed (1) hide show
  1. 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('split_docs.pkl', 'rb') as f:
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