Update app.py

app.py

@@ -107,7 +107,8 @@ def document_retrieval_mongodb(llm, prompt):
 
 def llm_chain(llm, prompt):
     llm_chain = LLMChain(llm = llm, prompt = LLM_CHAIN_PROMPT, verbose = False)
-    completion = llm_chain.run({"question": prompt})
+    #completion = llm_chain.run({"question": prompt})
+    completion = llm_chain.generate([{"question": prompt}])
     return completion, llm_chain
 
 def rag_chain(llm, prompt, db):