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)
|