vinny4 commited on
Commit
e712e61
·
1 Parent(s): 3586d61

added refiner in upload pdf

Browse files
Files changed (1) hide show
  1. 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)