Update app.py

Try inference instead of generate_response

app.py

@@ -9,7 +9,6 @@ from langchain_community.document_loaders import PyPDFLoader
 
 from langchain_text_splitters import RecursiveCharacterTextSplitter
 
-#from langchain_ollama import OllamaEmbeddings
 from langchain.embeddings import HuggingFaceEmbeddings
 
 import faiss
@@ -83,6 +82,11 @@ def get_pdf(uploaded_file):
     docs = loader.load()
     return docs
 
+def inference(chain, input_query):
+    """Invoke the processing chain with the input query."""
+    result = chain.invoke(input_query)
+    return result
+
 
 def main() -> None:
 
@@ -155,7 +159,12 @@ def main() -> None:
 
     if submit and text:
         with st.spinner("Generating response..."):
-            response = generate_response(rag_chain, text)
+            # Ken (12/11/2024): modify start
+            #response = generate_response(rag_chain, text)
+            ##retriever = configure_retriever(pdf_loader)
+            ##chain = create_chain(retriever, prompt, model)
+            response = inference(rag_chain, input_query)
+            # Ken (12/11/2024): modify end
             st.session_state['chat_history'].append({"user": text, "ollama": response})
             st.write(response)