Spaces:
Sleeping
Sleeping
added refiner in upload pdf
Browse files- src/pipeline.py +2 -0
src/pipeline.py
CHANGED
|
@@ -115,6 +115,8 @@ class ChatPipeline:
|
|
| 115 |
if not pdf_path:
|
| 116 |
raise ValueError("pdf_path must be provided to setup the pipeline.")
|
| 117 |
|
|
|
|
|
|
|
| 118 |
documents = extract_text_from_pdf(pdf_path)
|
| 119 |
|
| 120 |
preprocessed_docs = self._preprocess_docs(documents)
|
|
|
|
| 115 |
if not pdf_path:
|
| 116 |
raise ValueError("pdf_path must be provided to setup the pipeline.")
|
| 117 |
|
| 118 |
+
self.query_refiner = QueryRefiner()
|
| 119 |
+
|
| 120 |
documents = extract_text_from_pdf(pdf_path)
|
| 121 |
|
| 122 |
preprocessed_docs = self._preprocess_docs(documents)
|