notabaka commited on
Commit
3464d30
·
1 Parent(s): b21d556
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ from PyPDF2 import PdfReader
4
  import openpyxl
5
  from langchain.text_splitter import RecursiveCharacterTextSplitter
6
  from langchain.embeddings import GooglePalmEmbeddings
7
- from langchain.llms import GooglePalm
8
  from langchain.vectorstores import FAISS
9
  from langchain.chains import ConversationalRetrievalChain
10
  from langchain.memory import ConversationBufferMemory
@@ -41,7 +41,7 @@ def get_vector_store(text_chunks):
41
  return vector_store
42
 
43
  def get_conversational_chain(vector_store):
44
- llm = GooglePalm()
45
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
46
  conversation_chain = ConversationalRetrievalChain.from_llm(llm=llm, retriever=vector_store.as_retriever(), memory=memory)
47
  return conversation_chain
 
4
  import openpyxl
5
  from langchain.text_splitter import RecursiveCharacterTextSplitter
6
  from langchain.embeddings import GooglePalmEmbeddings
7
+ from langchain.llms import HuggingFaceTransformers # Updated import
8
  from langchain.vectorstores import FAISS
9
  from langchain.chains import ConversationalRetrievalChain
10
  from langchain.memory import ConversationBufferMemory
 
41
  return vector_store
42
 
43
  def get_conversational_chain(vector_store):
44
+ llm = HuggingFaceTransformers(model_name="HanNayeoniee/LHK_DPO_v1")
45
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
46
  conversation_chain = ConversationalRetrievalChain.from_llm(llm=llm, retriever=vector_store.as_retriever(), memory=memory)
47
  return conversation_chain