Update pages/bot.py

pages/bot.py

@@ -5,6 +5,7 @@ from langchain.text_splitter import CharacterTextSplitter
 from langchain.document_loaders import DirectoryLoader, PyPDFLoader
 import os
 from PyPDF2 import PdfReader
+from transformers import pipeline
 
 ###########
 #pip install faiss-cpu
@@ -80,6 +81,12 @@ def main():
     )
     if user_question:
         st.text(retrieved_docs[0].page_content)
+        qa_pipeline = pipeline("question-answering", model="hkunlp/instructor-base", tokenizer="hkunlp/instructor-base")
+        context=retrieved_docs[0].page_content
+        question=user_question
+        answer = qa_pipeline(question=question, context=context)
+        st.text("FORMATIERTE ANTWORT:")
+        st.text(answer)
                   # bei incoming pdf
 
     #vectorstore_DB=get_vectorstore()        # bei Abfrage durch Chatbot