Update app.py

app.py

@@ -150,7 +150,7 @@ if st.button("Submit Query"):
         custom_llm = CustomLanguageModel()
 
         # Extract context from the retriever
-        retrieved_docs = retriever.retrieve(query)
+        retrieved_docs = retriever.get_relevant_documents(query)
         context = format_docs(retrieved_docs)
 
         rag_chain = (