Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -74,7 +74,7 @@ embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwa
|
|
74 |
def predict(query):
|
75 |
persist_directory="./chroma_db4/"
|
76 |
db2 = Chroma(persist_directory=persist_directory, embedding_function=embeddings)
|
77 |
-
db2.get(include=[
|
78 |
docs = db2.similarity_search(query)
|
79 |
#context=docs[0]
|
80 |
context=docs[0].page_content
|
|
|
74 |
def predict(query):
|
75 |
persist_directory="./chroma_db4/"
|
76 |
db2 = Chroma(persist_directory=persist_directory, embedding_function=embeddings)
|
77 |
+
db2.get(include=["embeddings","metadatas","documents"])
|
78 |
docs = db2.similarity_search(query)
|
79 |
#context=docs[0]
|
80 |
context=docs[0].page_content
|