Update app.py

app.py

@@ -6,7 +6,7 @@ from langchain_community.vectorstores import FAISS
 from langchain.memory import ConversationBufferMemory
 from langchain.chains import ConversationalRetrievalChain
 from htmlTemplates import css, bot_template, user_template
-from langchain_community.llms import HuggingFaceEndpoint
+from langchain_community.llms import HuggingFaceHub
 import os
 from langchain_huggingface import HuggingFaceEmbeddings
 
@@ -41,11 +41,10 @@ def get_vectorstore(text_chunks):
 def get_conversation_chain(vectorstore):
     os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.getenv("HUGGINGFACEHUB_API_TOKEN", "")
 
-    llm = HuggingFaceEndpoint(
+    llm = HuggingFaceHub(
         repo_id="google/flan-t5-base",
         huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
-        temperature=0.5,
-        max_length=512
+        model_kwargs={"temperature": 0.5, "max_length": 512}
     )
 
     memory = ConversationBufferMemory(
@@ -61,6 +60,7 @@ def get_conversation_chain(vectorstore):
     return conversation_chain
 
 
+
 def handle_userinput(user_question):
     if st.session_state.conversation is None:
         st.warning("Please upload PDFs and click 'Process' before asking questions.")