amqdn commited on
Commit
b2a9886
1 Parent(s): 4b5c989

bugfix setup

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,10 +21,11 @@ def get_api_key(input_1, input_2):
21
  return True
22
 
23
  def setup_chain():
24
- global embeddings, docsearch, chain
25
  embeddings = OpenAIEmbeddings()
26
  docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))]).as_retriever()
27
  chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
 
28
 
29
  def make_inference(query):
30
  if not setup_complete:
 
21
  return True
22
 
23
  def setup_chain():
24
+ global embeddings, docsearch, chain, setup_complete
25
  embeddings = OpenAIEmbeddings()
26
  docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))]).as_retriever()
27
  chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
28
+ setup_complete = True
29
 
30
  def make_inference(query):
31
  if not setup_complete: