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]
|