Yoxas commited on
Commit
124c404
1 Parent(s): ce0825e

Update src/pdfchatbot.py

Browse files
Files changed (1) hide show
  1. src/pdfchatbot.py +2 -2
src/pdfchatbot.py CHANGED
@@ -122,7 +122,7 @@ class PDFChatBot:
122
  self.create_organic_pipeline()
123
  #self.create_chain()
124
  @spaces.GPU(duration=120)
125
- def generate_response(self, history, query, file,chunk_size,chunk_overlap_percentage,model_temperature,max_chunks_in_context):
126
 
127
  self.chunk_size = chunk_size
128
  self.overlap_percentage = chunk_overlap_percentage
@@ -144,7 +144,7 @@ class PDFChatBot:
144
  history[-1][-1] += char
145
  return history,""
146
 
147
- def render_file(self, file,chunk_size,chunk_overlap_percentage,model_temperature,max_chunks_in_context):
148
  print(chunk_size)
149
  doc = fitz.open(file.name)
150
  page = doc[self.page]
 
122
  self.create_organic_pipeline()
123
  #self.create_chain()
124
  @spaces.GPU(duration=120)
125
+ def generate_response(self, history, query, file, chunk_size, chunk_overlap_percentage, model_temperature, max_chunks_in_context):
126
 
127
  self.chunk_size = chunk_size
128
  self.overlap_percentage = chunk_overlap_percentage
 
144
  history[-1][-1] += char
145
  return history,""
146
 
147
+ def render_file(self, file, chunk_size, chunk_overlap_percentage, model_temperature, max_chunks_in_context):
148
  print(chunk_size)
149
  doc = fitz.open(file.name)
150
  page = doc[self.page]