Update pages/ChatPDF_Reader.py

pages/ChatPDF_Reader.py

@@ -30,8 +30,10 @@ def get_data(query):
     db = Chroma(persist_directory=chromadbpath, embedding_function=embeddings)
     retriever = db.as_retriever(search_type="mmr", search_kwargs={'k': 2})
 
-    llm = HuggingFaceHub(huggingfacehub_api_token=hf_token, 
-                        repo_id=llm_repo_id, model_kwargs={"temperature":0.2, "max_new_tokens":50})
+    """llm = HuggingFaceHub(huggingfacehub_api_token=hf_token, 
+                        repo_id=llm_repo_id, model_kwargs={"temperature":0.2, "max_new_tokens":50})"""
+    
+    llm = HuggingFaceHub(repo_id=llm_repo_id, model_kwargs={"temperature":0.2, "max_new_tokens":50})
 
     # Create the Conversational Retrieval Chain
     qa_chain = ConversationalRetrievalChain.from_llm(llm, retriever,return_source_documents=True)