Update app.py

app.py

@@ -3,7 +3,7 @@ from langchain_community.vectorstores import FAISS
 from langchain_community.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.embeddings import HuggingFaceEmbeddings
-from langchain_community.llms import HuggingFaceEndpoint
+from langchain.llms import HuggingFaceHub
 from langchain.chains import ConversationalRetrievalChain
 from langchain.memory import ConversationBufferMemory
 import os
@@ -67,13 +67,12 @@ def create_db(splits):
     return vectordb
 
 def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db):
-    """Initialize the LLM chain with correct parameter names"""
-    llm = HuggingFaceEndpoint(
-        endpoint_url="https://api-inference.huggingface.co/models/" + llm_model,
-        task="text-generation",
+    """Initialize the LLM chain with correct parameters"""
+    llm = HuggingFaceHub(
+        repo_id=llm_model,
         model_kwargs={
             "temperature": float(temperature),
-            "max_length": int(max_tokens),
+            "max_new_tokens": int(max_tokens),
             "top_k": int(top_k)
         },
         huggingfacehub_api_token=api_token