Update pages/bot.py
Browse files- pages/bot.py +1 -1
pages/bot.py
CHANGED
@@ -81,7 +81,7 @@ def main():
|
|
81 |
)
|
82 |
if user_question:
|
83 |
st.text(retrieved_docs[0].page_content)
|
84 |
-
qa_pipeline = pipeline("question-answering", model="
|
85 |
context=retrieved_docs[0].page_content
|
86 |
question=user_question
|
87 |
answer = qa_pipeline(question=question, context=context)
|
|
|
81 |
)
|
82 |
if user_question:
|
83 |
st.text(retrieved_docs[0].page_content)
|
84 |
+
qa_pipeline = pipeline("question-answering", model="tiiuae/falcon-40b", tokenizer="tiiuae/falcon-40b")
|
85 |
context=retrieved_docs[0].page_content
|
86 |
question=user_question
|
87 |
answer = qa_pipeline(question=question, context=context)
|