Update app.py

app.py

@@ -93,7 +93,7 @@ def invoke(openai_api_key, rag, prompt):
             ##                                        return_source_documents = True)
             ##result = rag_chain({"query": prompt})
             ##result = result["result"]
-            result = document_retrieval_chroma(llm)
+            result = document_retrieval_chroma(llm, prompt)
         else:
             chain = LLMChain(llm = llm, prompt = LLM_CHAIN_PROMPT)
             result = chain.run({"question": prompt})