talexm commited on
Commit
d8007de
1 Parent(s): ed26242
Files changed (1) hide show
  1. rag_sec/document_search_system.py +1 -1
rag_sec/document_search_system.py CHANGED
@@ -184,7 +184,7 @@ def main():
184
  )
185
 
186
  # Load documents into the retriever
187
- system.retriever.load_documents(data_dir)
188
  print("Documents successfully loaded.")
189
 
190
  return system
 
184
  )
185
 
186
  # Load documents into the retriever
187
+ system.retriever.load_documents()
188
  print("Documents successfully loaded.")
189
 
190
  return system