Update pages/bot.py

pages/bot.py

@@ -100,13 +100,30 @@ def main():
         st.text(answer)
         st.text(type(answer))
 
+        # Erstelle die Question Answering-Pipeline für Deutsch
+        qa_pipeline = pipeline("question-answering", model="deutsche-telekom/bert-multi-english-german-squad2", tokenizer="deutsche-telekom/bert-multi-english-german-squad2")
+
+        # Frage beantworten
+        answer = qa_pipeline(question=question, context=context)
+        
+        # Gib die Antwort aus
+        st.text("Basisantwort:", answer["answer"])
+        st.text(answer)
+        
+        
+        generator = pipeline('text-generation', model = 'jphme/Llama-2-13b-chat-german')
+        generator(answer, max_length = 30, num_return_sequences=3)
+        st.text("Generierte Erweiterung:")
+        st.text(generator)
+            
+        """
         #IDEE Retriever erweitern
-        template = """Answer the question based only on the following context:
+        template = Answer the question based only on the following context:
 
         {context}
         
         Question: {question}
-        """
+        
         prompt = ChatPromptTemplate.from_template(template)
         model = AutoModel.from_pretrained("hkunlp/instructor-base")
         
@@ -124,6 +141,7 @@ def main():
 
         ausgabetext = chain.invoke(user_question)
         st.text(ausgabetext)
+        """