Update pages/bot.py

pages/bot.py

@@ -81,10 +81,10 @@ def main():
     )
     if user_question:
         st.text(retrieved_docs[0].page_content)
-        qa_pipeline = pipeline("question-answering", model="tiiuae/falcon-40b", tokenizer="tiiuae/falcon-40b")
         context=retrieved_docs[0].page_content
         question=user_question
-        answer = qa_pipeline(question=question, context=context)
+        generator = pipeline('text-generation', model = 'gpt2')
+        answer = generator(context, max_length = 30, num_return_sequences=3)
         st.text("FORMATIERTE ANTWORT:")
         st.text(answer)
                   # bei incoming pdf