Update pages/bot.py

pages/bot.py

@@ -96,6 +96,7 @@ def main():
         st.text(user_question)
         context=""+retrieved_docs[0].page_content+retrieved_docs[1].page_content+retrieved_docs[3].page_content
         context=context.replace("\n", " ")  
+        context=context.replace("- ", "")  
         st.text("Das ist der Textausschnitt der durch den Retriever herausgesucht wird:")
         st.text(context)
 
@@ -112,9 +113,9 @@ def main():
         #st.text(answer)
 
         #Die Basisantwort müsste man jetzt ausformulieren
-        text2text_generator = pipeline("text2text-generation")
+        text2text_generator = pipeline("text2text-generation", model="google/flan-t5-base")
         #newText=text2text_generator(question=question, context=answer)
-        newText=text2text_generator("question: "+question+ " context: " + answer["answer"])
+        newText=text2text_generator("question: "+question+ " answer: " + answer["answer"])
         st.text(newText)
 
 if __name__ == '__main__':