Spaces:
Runtime error
Runtime error
gkrthk
commited on
Commit
•
72b8502
1
Parent(s):
b4493c7
update
Browse files- 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("
|
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:
|