chukbert commited on
Commit
db87dc4
·
verified ·
1 Parent(s): 073f197

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -31,7 +31,14 @@ def process_pdf(uploaded_file):
31
 
32
  def create_vectorstore(splits):
33
  embeddings = OpenAIEmbeddings()
34
- vectorstore = Chroma.from_documents(splits, embeddings)
 
 
 
 
 
 
 
35
  retriever = vectorstore.as_retriever()
36
  return retriever
37
 
 
31
 
32
  def create_vectorstore(splits):
33
  embeddings = OpenAIEmbeddings()
34
+ vectorstore = Chroma.from_documents(
35
+ splits,
36
+ embeddings,
37
+ client_settings=Settings(
38
+ chroma_db_impl="duckdb+parquet", # Use DuckDB in-memory for Chroma
39
+ persist_directory=None # No need for persistent storage
40
+ )
41
+ )
42
  retriever = vectorstore.as_retriever()
43
  return retriever
44