Update app.py
Browse files
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='
|
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)
|