Pavan178 commited on
Commit
88d5586
·
verified ·
1 Parent(s): 0bd2d04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ class AdvancedPdfChatbot:
22
  self.embeddings = OpenAIEmbeddings()
23
  self.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
24
  self.llm = ChatOpenAI(temperature=0.7,model_name='gpt-4o',max_tokens=2048,top_p = 0.7)
25
- self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
26
  self.qa_chain = None
27
  self.pdf_path = None
28
  self.template = """
 
22
  self.embeddings = OpenAIEmbeddings()
23
  self.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
24
  self.llm = ChatOpenAI(temperature=0.7,model_name='gpt-4o',max_tokens=2048,top_p = 0.7)
25
+ self.memory = ConversationBufferMemory(memory_key="chat_history", output_key='answer',return_messages=True)
26
  self.qa_chain = None
27
  self.pdf_path = None
28
  self.template = """