Chris Alexiuk commited on
Commit
28e9db0
·
1 Parent(s): 7bfdf49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} f
18
  chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
19
 
20
  def make_inference(query):
 
21
  chain.run(input_documents=docs, question=query)
22
 
23
  if __name__ == "__main__":
 
18
  chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
19
 
20
  def make_inference(query):
21
+ docs = docsearch.get_relevant_documents(query)
22
  chain.run(input_documents=docs, question=query)
23
 
24
  if __name__ == "__main__":