Update src/pdfchatbot.py
Browse files- 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]
|