FlavioBF commited on
Commit
bf9b41c
1 Parent(s): d5af638

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=[include=["embeddings","metadatas","documents"])
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