tamizh-me commited on
Commit
b51b347
·
verified ·
1 Parent(s): 2fe5985

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ llm = ChatAnthropic(model='claude-2',
15
  temperature=0)
16
 
17
  # Load the persisted Chroma database
18
- persist_directory='/Electric_Machinerydb'
19
  embeddings = embeddings
20
  chroma_db = Chroma(persist_directory=persist_directory, embedding_function=embeddings)
21
  retriever = chroma_db.as_retriever(search_kwargs={"k": 3}, return_source_documents=True)
 
15
  temperature=0)
16
 
17
  # Load the persisted Chroma database
18
+ persist_directory = os.path.expanduser('~/Electric_Machinerydb')
19
  embeddings = embeddings
20
  chroma_db = Chroma(persist_directory=persist_directory, embedding_function=embeddings)
21
  retriever = chroma_db.as_retriever(search_kwargs={"k": 3}, return_source_documents=True)