Adipta commited on
Commit
4f43089
·
verified ·
1 Parent(s): 63dccee

Update pdfchatbot.py

Browse files
Files changed (1) hide show
  1. pdfchatbot.py +1 -1
pdfchatbot.py CHANGED
@@ -132,7 +132,7 @@ class PDFChatBot:
132
 
133
  result = self.chain({"question": query, 'chat_history': self.chat_history}, return_only_outputs=True)
134
  self.chat_history.append((query, result["answer"]))
135
- self.page = list(result['source_documents'][0])[1][1]['page']
136
 
137
  for char in result['answer']:
138
  history[-1][-1] += char
 
132
 
133
  result = self.chain({"question": query, 'chat_history': self.chat_history}, return_only_outputs=True)
134
  self.chat_history.append((query, result["answer"]))
135
+ self.page = 0
136
 
137
  for char in result['answer']:
138
  history[-1][-1] += char