gkrthk commited on
Commit
72b8502
1 Parent(s): b4493c7
Files changed (1) hide show
  1. confluence_qa.py +1 -1
confluence_qa.py CHANGED
@@ -17,7 +17,7 @@ class ConfluenceQA:
17
 
18
  # tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
19
  # model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
20
- pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
21
  self.llm = HuggingFacePipeline(pipeline = pipe,model_kwargs={"temperature": 0})
22
 
23
  def store_in_vector_db(self) -> None:
 
17
 
18
  # tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
19
  # model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
20
+ pipe = pipeline("question-answering", model=model, tokenizer=tokenizer)
21
  self.llm = HuggingFacePipeline(pipeline = pipe,model_kwargs={"temperature": 0})
22
 
23
  def store_in_vector_db(self) -> None: