ramortegui commited on
Commit
fd88fa4
·
unverified ·
1 Parent(s): e85540e
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -94,6 +94,7 @@ docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} f
94
  chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
95
 
96
  def query(query):
 
97
  return chain.run(input_documents=docs, question=query)
98
 
99
 
 
94
  chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
95
 
96
  def query(query):
97
+ docs = docsearch.get_relevant_documents(query)
98
  return chain.run(input_documents=docs, question=query)
99
 
100