ErikH commited on
Commit
d9e7248
1 Parent(s): 5b68c17

Update pages/bot.py

Browse files
Files changed (1) hide show
  1. 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="hkunlp/instructor-base", tokenizer="hkunlp/instructor-base")
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)