VcRlAgent commited on
Commit
eb08d30
·
1 Parent(s): 58d0790

Fix LlamaIndex SimpleDir

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def load_llamaindex_stack(model_id: str, max_new_tokens: int, temperature: float
83
  #Settings.llm = llm
84
 
85
  # Load small docs (data/notes.txt)
86
- docs = SimpleDirectoryReader(input_dir=["data"]).load_data()
87
  index = VectorStoreIndex.from_documents(docs)
88
  query_engine = index.as_query_engine(similarity_top_k=3)
89
  return query_engine
 
83
  #Settings.llm = llm
84
 
85
  # Load small docs (data/notes.txt)
86
+ docs = SimpleDirectoryReader(input_dir="data").load_data()
87
  index = VectorStoreIndex.from_documents(docs)
88
  query_engine = index.as_query_engine(similarity_top_k=3)
89
  return query_engine